accelerate directlingam with cuda implementation

aknvictor · aknvictor · commit 6039bba66060 · 2024-03-01T20:12:54.000-05:00
Signed-off-by: Victor &lt;viktour19@gmail.com&gt;
diff --git a/tests/TestDirectLiNGAMfast.py b/tests/TestDirectLiNGAMfast.py
@@ -34,7 +34,7 @@ def test_DirectLiNGAM(self):
 
         cuda = get_cuda_version()
         if cuda:
-            model = lingam.DirectLiNGAM()
+            model = lingam.DirectLiNGAM(measure='pwling_fast')
             model.fit(X)
 
             print(model.causal_order_)
