fix dowhy_ not existance issue

py-why · Nov 21, 2024 · ada34cc · ada34cc
1 parent 54275bc
commit ada34cc
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/econml/dowhy.py b/econml/dowhy.py
@@ -231,9 +231,13 @@ def __getattr__(self, attr):
         # don't proxy special methods
         if attr.startswith('__'):
             raise AttributeError(attr)
-        elif attr in ['_cate_estimator', 'dowhy_',
-                      'identified_estimand_', 'estimate_']:
-            return super().__getattr__(attr)
+        elif attr == "dowhy_":
+            if "dowhy_" not in dir(self):
+                raise RuntimeError("Please call DoWhyWrapper.fit first before any other operations")
+            else:
+                return self.dowhy_
+        elif attr in ['_cate_estimator', 'identified_estimand_', 'estimate_']:
+            return getattr(self, attr)
         elif attr.startswith('dowhy__'):
             return getattr(self.dowhy_, attr[len('dowhy__'):])
         elif hasattr(self.estimate_._estimator_object, attr):