Skip to content

Commit a15f00b

Browse files
committed
Fix typing for RWSMock
1 parent 3c6b5c6 commit a15f00b

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/abb_robot_client/rws_mock.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -236,10 +236,12 @@ def get_mechunits(self) -> List[str]:
236236
return ["ROB_1"]
237237

238238
# Geometry helpers
239-
def get_jointtarget(self) -> JointTarget:
239+
def get_jointtarget(self, mechunit: str = "ROB_1") -> JointTarget:
240240
return JointTarget(robax=np.zeros(6), extax=np.zeros(6))
241241

242-
def get_robtarget(self) -> RobTarget:
242+
def get_robtarget(
243+
self, mechunit: str = "ROB_1", tool: str = "tool0", wobj: str = "wobj0", coordinate: str = "Base"
244+
) -> RobTarget:
243245
return RobTarget(
244246
trans=np.zeros(3),
245247
rot=np.array([1.0, 0.0, 0.0, 0.0]),

0 commit comments

Comments
 (0)