accelerate directlingam with cuda implementation

py-why · Mar 2, 2024 · ec7e096 · ec7e096
1 parent d83a888
commit ec7e096
Showing 1 changed file with 1 addition and 1 deletion.
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_)