fix: typing

engnadeau · Aug 23, 2022 · 7cb5a3c · 7cb5a3c
1 parent f8e4d22
commit 7cb5a3c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/examples/basic_usage.py b/examples/basic_usage.py
@@ -3,6 +3,7 @@
 from pybotics.predefined_models import ur10
 from pybotics.robot import Robot
 from pybotics.tool import Tool
+import numpy as np
 
 
 def main() -> None:
@@ -16,11 +17,11 @@ def main() -> None:
 
     # add tool
     tool = Tool()
-    tool.position = [1, 2, 3]
+    tool.position = np.array([1, 2, 3])
     robot.tool = tool
 
     # set world frame
-    world_frame = vector_2_matrix([100, 200, 300, 0, 0, 0])
+    world_frame = vector_2_matrix(np.array([100, 200, 300, 0, 0, 0]))
     robot.world_frame = world_frame
 
     print(f"Robot: {robot}")