make f2 output an array

NLESC-JCER · Dec 5, 2024 · 91cfde4 · 91cfde4
1 parent 69bb397
commit 91cfde4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 67 deletions.
diff --git a/.github/workflows/build_osx.yml b/.github/workflows/build_osx.yml
diff --git a/qmctorch/utils/algebra_utils.py b/qmctorch/utils/algebra_utils.py
@@ -1,4 +1,5 @@
 import torch
+import numpy as np
 from scipy.special import factorial2 as f2
 
 def btrace(M):
@@ -53,7 +54,7 @@ def double_factorial(input):
         List: values of the double factorial
     """
     output = f2(input)
-    return [1 if o==0 else o for o in output]
+    return np.array([1 if o==0 else o for o in output])
 
 
 class BatchDeterminant(torch.autograd.Function):