fix in unit test

erdogant · Nov 10, 2023 · 924796b · 924796b
1 parent a1b3755
commit 924796b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bnlearn/tests/test_bnlearn.py b/bnlearn/tests/test_bnlearn.py
@@ -85,7 +85,8 @@ def sprinkler_dag():
 
 def test_sampling_bayes(sprinkler_dag):
     n = np.random.randint(10, 1000)
-    df = bn.sampling(sprinkler_dag, n=n, methodtype='bayes')
+    model = bn.import_DAG('sprinkler')
+    df = bn.sampling(model, n=n, methodtype='bayes')
     assert df.shape == (n, 4)