From be1a9bd67eb303d5756bd2568f9b2e5e4efb7151 Mon Sep 17 00:00:00 2001 From: Hongyi Zhou Date: Tue, 22 Nov 2022 10:46:39 +0100 Subject: [PATCH] update bc --- fancy_gym/black_box/black_box_wrapper.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/fancy_gym/black_box/black_box_wrapper.py b/fancy_gym/black_box/black_box_wrapper.py index 5b9a5b9..c85ed14 100644 --- a/fancy_gym/black_box/black_box_wrapper.py +++ b/fancy_gym/black_box/black_box_wrapper.py @@ -116,9 +116,12 @@ class BlackBoxWrapper(gym.ObservationWrapper): self.condition_pos = self.current_pos self.condition_vel = self.current_vel - bc_time = torch.as_tensor(bc_time, dtype=torch.float32) - self.condition_pos = torch.as_tensor(self.condition_pos, dtype=torch.float32) - self.condition_vel = torch.as_tensor(self.condition_vel, dtype=torch.float32) + # bc_time = torch.as_tensor(bc_time, dtype=torch.float32) + # self.condition_pos = torch.as_tensor(self.condition_pos, dtype=torch.float32) + # self.condition_vel = torch.as_tensor(self.condition_vel, dtype=torch.float32) + print("bc_time", bc_time) + print("self.condition_pos", self.condition_pos) + print("self.condition_vel", self.condition_vel) self.traj_gen.set_boundary_conditions(bc_time, self.condition_pos, self.condition_vel) # self.traj_gen.set_duration(duration, self.dt) self.traj_gen.set_duration(duration, self.dt)