From e30dd104144a8249de7d720f6c5c318298a12309 Mon Sep 17 00:00:00 2001
From: Roland Stevenson <rolandrmgservices@gmail.com>
Date: Mon, 11 Dec 2023 13:28:17 +0100
Subject: [PATCH] avoid importing *

---
 tests/test_dragonnet.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_dragonnet.py b/tests/test_dragonnet.py
index 14eaabf6..1d4e613d 100644
--- 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