fix(ingest/sagemaker): remove unsupported config (#10606)

This commit is contained in:
Harshal Sheth 2024-05-28 13:42:54 -07:00 committed by GitHub
parent 0fd451c517
commit 81ddfa1032
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -3,7 +3,6 @@ from typing import Dict, List, Optional, Union
from pydantic.fields import Field from pydantic.fields import Field
from datahub.configuration.source_common import PlatformInstanceConfigMixin
from datahub.ingestion.source.aws.aws_common import AwsSourceConfig from datahub.ingestion.source.aws.aws_common import AwsSourceConfig
from datahub.ingestion.source.state.stale_entity_removal_handler import ( from datahub.ingestion.source.state.stale_entity_removal_handler import (
StaleEntityRemovalSourceReport, StaleEntityRemovalSourceReport,
@ -14,7 +13,6 @@ from datahub.ingestion.source.state.stale_entity_removal_handler import (
class SagemakerSourceConfig( class SagemakerSourceConfig(
AwsSourceConfig, AwsSourceConfig,
PlatformInstanceConfigMixin,
StatefulIngestionConfigBase, StatefulIngestionConfigBase,
): ):
extract_feature_groups: Optional[bool] = Field( extract_feature_groups: Optional[bool] = Field(