kinematics refactor with mimic joint support

This commit is contained in:
Balakumar Sundaralingam
2024-04-03 18:42:01 -07:00
parent b481ee201a
commit 774dcfd609
60 changed files with 2177 additions and 810 deletions

View File

@@ -274,7 +274,7 @@ class ArmReacher(ArmBase, ArmReacherConfig):
current_fn = self._link_pose_costs[k]
if current_fn.enabled:
# get link pose
current_pose = link_poses[k]
current_pose = link_poses[k].contiguous()
current_pos = current_pose.position
current_quat = current_pose.quaternion

View File

@@ -417,6 +417,7 @@ class Goal(Sequence):
class RolloutConfig:
tensor_args: TensorDeviceType
sum_horizon: bool = False
sampler_seed: int = 1312
class RolloutBase:
@@ -436,7 +437,7 @@ class RolloutBase:
self.tensor_args,
up_bounds=self.action_bound_highs,
low_bounds=self.action_bound_lows,
seed=1312,
seed=self.sampler_seed,
)
@abstractmethod