pass on tensorflow ImportError

uber · Jan 11, 2024 · 8fc7abc · 8fc7abc
1 parent 34cddfc
commit 8fc7abc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_dragonnet.py b/tests/test_dragonnet.py
@@ -1,4 +1,7 @@
-from causalml.inference.tf import DragonNet
+try:
+    from causalml.inference.tf import DragonNet
+except ImportError:
+    pass
 from causalml.dataset.regression import simulate_nuisance_and_easy_treatment
 import shutil
 import pytest