Update basis_fd.py

Barry57 · Oct 27, 2024 · 46c0f25 · 46c0f25
1 parent 95716d8
commit 46c0f25
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/GENetLib/basis_fd.py b/GENetLib/basis_fd.py
@@ -39,9 +39,9 @@ def basis_fd(btype=None, rangeval=None, nbasis=None, params=None, dropind=None,
         btype = "power"
     else:
         btype = "unknown"
-    if nbasis <= 0:
+    if nbasis is not None and nbasis<= 0:
         raise ValueError("Argument nbasis is not positive.")
-    if round(nbasis) != nbasis:
+    if nbasis is not None and round(nbasis) != nbasis:
         raise ValueError("Argument nbasis is not an integer.")
     if quadvals is None:
         quadvals = []
@@ -141,4 +141,4 @@ def basis_fd(btype=None, rangeval=None, nbasis=None, params=None, dropind=None,
         }
         return basisobj
 
-
+