UM-ARM-Lab / pytorch_kinematics

Robot kinematics implemented in pytorch
MIT License
443 stars 42 forks source link

Minor issue due to mismatch of data types in transform3d.py #47

Open kidpaul94 opened 6 days ago

kidpaul94 commented 6 days ago

In line 749 of transform3d.py,

https://github.com/UM-ARM-Lab/pytorch_kinematics/blob/a457cda0111e3ed836988d2e7b6652d969aa26ec/src/pytorch_kinematics/transforms/transform3d.py#L749

It keeps causing an error when I try to visualize sdf using pv.draw_sdf_slice because a is a float tensor while b is a double tensor.

Just a heads up since I'm not sure how critical this is when using other functions in pytorch_kinematics.

I faced this issue when I tried to specify dtype of a chain class.