From 09fb69b68dace1616f232c62318267ef4c8d8b3c Mon Sep 17 00:00:00 2001 From: Onkar Ravgan Date: Wed, 19 Apr 2023 21:39:29 +0530 Subject: [PATCH] Fixed objectstore import (#11144) --- .../src/metadata/ingestion/source/metadata/metadata.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/ingestion/src/metadata/ingestion/source/metadata/metadata.py b/ingestion/src/metadata/ingestion/source/metadata/metadata.py index 1c74d463d3b..53bcb985a1a 100644 --- a/ingestion/src/metadata/ingestion/source/metadata/metadata.py +++ b/ingestion/src/metadata/ingestion/source/metadata/metadata.py @@ -33,10 +33,8 @@ from metadata.generated.schema.entity.services.connections.metadata.openMetadata ) from metadata.generated.schema.entity.services.databaseService import DatabaseService from metadata.generated.schema.entity.services.messagingService import MessagingService -from metadata.generated.schema.entity.services.objectstoreService import ( - ObjectStoreService, -) from metadata.generated.schema.entity.services.pipelineService import PipelineService +from metadata.generated.schema.entity.services.storageService import StorageService from metadata.generated.schema.entity.teams.team import Team from metadata.generated.schema.entity.teams.user import User from metadata.generated.schema.metadataIngestion.workflow import ( @@ -179,7 +177,7 @@ class MetadataSource(Source[Entity]): if self.service_connection.includeStorageServices: yield from self.fetch_entities( - entity_class=ObjectStoreService, + entity_class=StorageService, fields=["owner"], )