minor fix

datahub-project · Feb 24, 2025 · 5b35761 · 5b35761
1 parent 003f257
commit 5b35761
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/vertexai.py b/metadata-ingestion/src/datahub/ingestion/source/vertexai.py
@@ -9,7 +9,8 @@
     AutoMLTabularTrainingJob,
     AutoMLTextTrainingJob,
     AutoMLVideoTrainingJob,
-    Endpoint, TabularDataset, TextDataset,
+    Endpoint,
+    TabularDataset,
 )
 from google.cloud.aiplatform.datasets import _Dataset
 from google.cloud.aiplatform.models import Model, VersionInfo
@@ -33,7 +34,8 @@
     AuditStampClass,
     DataProcessInstanceInputClass,
     DataProcessInstancePropertiesClass,
-    TimeStampClass, DatasetPropertiesClass,
+    DatasetPropertiesClass,
+    TimeStampClass,
 )
 from datahub.metadata.com.linkedin.pegasus2avro.dataset import DatasetProperties
 from datahub.metadata.schema_classes import (