From b1f4ac84f19d3ea17dddb6141071b67633975a8e Mon Sep 17 00:00:00 2001 From: Brian Delhaisse Date: Wed, 20 Nov 2019 04:22:04 +0100 Subject: [PATCH] fix roll in converter.py --- pyrobolearn/utils/converter.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pyrobolearn/utils/converter.py b/pyrobolearn/utils/converter.py index e297a6d..1556c3e 100644 --- a/pyrobolearn/utils/converter.py +++ b/pyrobolearn/utils/converter.py @@ -251,7 +251,7 @@ class QuaternionNumpyConverter(TypeConverter): if isinstance(data, self.to_type): return data elif isinstance(data, self.from_type): - return np.quaternion(roll(data.ravel().tolist(), -self.convention)) + return np.quaternion(*roll(data.ravel().tolist(), -self.convention)) else: raise TypeError("Type not known: {}".format(type(data))) @@ -301,7 +301,7 @@ class QuaternionPyTorchConverter(TypeConverter): if isinstance(data, self.to_type): return data elif isinstance(data, self.from_type): - return np.quaternion(roll(data.view(-1).data.tolist(), -self.convention)) + return np.quaternion(*roll(data.view(-1).data.tolist(), -self.convention)) else: raise TypeError("Type not known: {}".format(type(data)))