Force int64 to python int

GreshamLab · Oct 17, 2023 · ec18685 · ec18685
1 parent 5714cd9
commit ec18685
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/supirfactor_dynamical/_utils/_loader.py b/supirfactor_dynamical/_utils/_loader.py
@@ -55,6 +55,10 @@
     4: 'tanh'
 }
 
+_FORCE_UNIT = {
+    'decay_k': int
+}
+
 
 def read(
     file_name,
@@ -103,6 +107,10 @@ def read(
             for k in _state_args
         }
 
+    for k, func in _FORCE_UNIT.items():
+        if k in kwargs:
+            kwargs[k] = func(kwargs[k])
+
     # Get the prior and pop out the prior
     # size variables
     try: