avoid importing *

uber · Dec 11, 2023 · e30dd10 · e30dd10
1 parent 98f08db
commit e30dd10
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_dragonnet.py b/tests/test_dragonnet.py
@@ -1,5 +1,5 @@
 from causalml.inference.tf import DragonNet
-from causalml.dataset.regression import *
+from causalml.dataset.regression import simulate_nuisance_and_easy_treatment
 import shutil