fixed np.int64 datatype mismatch with randomQuaternion and randomMatrix

Eric-Vignola · Apr 11, 2021 · 1c15ce0 · 1c15ce0
1 parent 794db7f
commit 1c15ce0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/transforms.py b/transforms.py
@@ -1314,14 +1314,14 @@ def randomEuler(n, seed=None):
 def randomQuaternion(n, seed=None):
     """ Computes a list of random quaternions qi,qj,qk,qw
     """
-    return _eulerToQuaternion(randomEuler(n=n, seed=seed),np.array([0]))
+    return _eulerToQuaternion(randomEuler(n=n, seed=seed),np.array([0]).astype(np.int64))
 
 
 
 def randomMatrix(n, seed=None, random_position=False):
     """ Computes a list of random 4x4 rotation matrices
     """
-    M = _eulerToMatrix(randomEuler(n=n, seed=seed),np.array([0]))
+    M = _eulerToMatrix(randomEuler(n=n, seed=seed),np.array([0]).astype(np.int64))
     if random_position:
         M[:,3,:3] = 1 - np.random.random((n,3,))*2