Added test

StatMixedML · Aug 2, 2023 · 7ceed37 · 7ceed37
1 parent d4e5299
commit 7ceed37
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/test_distribution_utils/test_stabilize_derivative.py b/test/test_distribution_utils/test_stabilize_derivative.py
@@ -1,5 +1,6 @@
 from ..utils import BaseTestClass
-from xgboostlss.model import *
+import torch
+
 
 
 class TestClass(BaseTestClass):