fix lint

datahub-project · Sep 4, 2023 · c21d66a · c21d66a
1 parent a91413d
commit c21d66a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/transformer/extract_ownership_from_tags.py b/metadata-ingestion/src/datahub/ingestion/transformer/extract_ownership_from_tags.py
@@ -28,7 +28,7 @@ class ExtractOwnersFromTagsConfig(TransformerSemanticsConfigModel):
 def get_owner_type(owner_type_str: str) -> Optional[OwnershipTypeClass]:
     for item in dir(OwnershipTypeClass):
         if str(item) == owner_type_str:
-            return item
+            return getattr(OwnershipTypeClass, item)
     return OwnershipTypeClass.CUSTOM
 
 
@@ -46,7 +46,7 @@ def __init__(self, config: ExtractOwnersFromTagsConfig, ctx: PipelineContext):
     @classmethod
     def create(
         cls, config_dict: dict, ctx: PipelineContext
-    ) -> "ExtractOwnersFromTagsConfig":
+    ) -> "ExtractOwnersFromTagsTransformer":
         config = ExtractOwnersFromTagsConfig.parse_obj(config_dict)
         return cls(config, ctx)