accelerate directlingam with cuda implementation

aknvictor · aknvictor · commit ec7e096cd231 · 2024-03-01T20:11:38.000-05:00
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_)
