Add explicit casting to avoid windows error

AdaptiveMotorControlLab · Oct 27, 2024 · e02b9a8 · e02b9a8
1 parent 5b24721
commit e02b9a8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/cebra/data/datasets.py b/cebra/data/datasets.py
@@ -108,6 +108,10 @@ def _to_tensor(
                     f"Array has type {array.dtype} instead of {check_dtype}.")
         if cebra_helper._is_floating(array):
             array = array.float()
+        if cebra_helper._is_integer(array):
+            # NOTE(stes): Required for standardizing number format on
+            # windows machines.
+            array = array.long()
         return array
 
     @property