fix(ingest): avoid mutable defaults in powerbi dataclass (#8609)

datahub-project · Aug 24, 2023 · 03ea29c · 03ea29c
1 parent cc8b007
commit 03ea29c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/api/registry.py b/metadata-ingestion/src/datahub/ingestion/api/registry.py
@@ -127,7 +127,7 @@ def _ensure_not_lazy(self, key: str) -> Union[Type[T], Exception]:
             plugin_class = import_path(path)
             self.register(key, plugin_class, override=True)
             return plugin_class
-        except (AssertionError, ImportError) as e:
+        except Exception as e:
             self.register_disabled(key, e, override=True)
             return e
 

diff --git a/metadata-ingestion/src/datahub/ingestion/source/powerbi/rest_api_wrapper/data_classes.py b/metadata-ingestion/src/datahub/ingestion/source/powerbi/rest_api_wrapper/data_classes.py
@@ -1,3 +1,4 @@
+import dataclasses
 from dataclasses import dataclass
 from enum import Enum
 from typing import Any, Dict, List, Optional, Union
@@ -105,7 +106,7 @@ class Measure:
     dataType: str = "measure"
     datahubDataType: Union[
         BooleanTypeClass, DateTypeClass, NullTypeClass, NumberTypeClass, StringTypeClass
-    ] = NullTypeClass()
+    ] = dataclasses.field(default_factory=NullTypeClass)
     description: Optional[str] = None