diff --git a/metadata-ingestion/src/datahub/ingestion/source/feast.py b/metadata-ingestion/src/datahub/ingestion/source/feast.py index 835c64770f2b6..ff4e9a48fe7b2 100644 --- a/metadata-ingestion/src/datahub/ingestion/source/feast.py +++ b/metadata-ingestion/src/datahub/ingestion/source/feast.py @@ -33,7 +33,6 @@ ) from datahub.ingestion.api.source import Source, SourceReport from datahub.ingestion.api.workunit import MetadataWorkUnit - from datahub.metadata.com.linkedin.pegasus2avro.common import MLFeatureDataType from datahub.metadata.com.linkedin.pegasus2avro.metadata.snapshot import ( MLFeatureSnapshot, @@ -47,11 +46,11 @@ MLFeaturePropertiesClass, MLFeatureTablePropertiesClass, MLPrimaryKeyPropertiesClass, - StatusClass, - TagAssociationClass, OwnerClass, OwnershipClass, OwnershipTypeClass, + StatusClass, + TagAssociationClass, ) # FIXME: ValueType module cannot be used as a type