Merge pull request #242 from pollytur/core3d_bn_fix

Factorial fix
sinzlab · Jun 6, 2024 · a2055cd · a2055cd
2 parents 5947994 + 2da2835
commit a2055cd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neuralpredictors/layers/hermite.py b/neuralpredictors/layers/hermite.py
@@ -65,7 +65,7 @@ def hermite_2d(N, npts, xvalmax=None):
     for i, (mui, nui, desci) in enumerate(zip(mu, nu, desc)):
         radvals = polyval(radsq, hermite_coefficients(mui, nui))
         basis = gaussian * (radsq ** (mui / 2)) * radvals * np.exp(1j * mui * theta)
-        basis /= np.sqrt(2 ** (mui + 2 * nui) * pi * math.factorial(mui + nui) * math.factorial(nui))
+        basis /= np.sqrt(2 ** (mui + 2 * nui) * pi * math.factorial(int(mui + nui)) * math.factorial(int(nui)))
         if desci == "z":
             H[i] = basis.real / np.sqrt(2)
         elif desci == "r":