accelerate directlingam with cuda implementation

Signed-off-by: Victor <viktour19@gmail.com>
py-why · Mar 2, 2024 · 6039bba · 6039bba
1 parent 2ee618b
commit 6039bba
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_)