From a219f90d7230d0f0ba88210cfa586b464bfd73a1 Mon Sep 17 00:00:00 2001
From: Hongyi Zhou <wolfgangzhou1996@outlook.com>
Date: Tue, 22 Nov 2022 11:13:08 +0100
Subject: [PATCH] shorter duration

---
 fancy_gym/black_box/black_box_wrapper.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/fancy_gym/black_box/black_box_wrapper.py b/fancy_gym/black_box/black_box_wrapper.py
index c85ed14..3732863 100644
--- a/fancy_gym/black_box/black_box_wrapper.py
+++ b/fancy_gym/black_box/black_box_wrapper.py
@@ -99,7 +99,8 @@ class BlackBoxWrapper(gym.ObservationWrapper):
 
     def get_trajectory(self, action: np.ndarray) -> Tuple:
         # duration = self.duration
-        duration = self.duration - self.current_traj_steps * self.dt
+        # duration = self.duration - self.current_traj_steps * self.dt
+        duration = action[0]
         if self.learn_sub_trajectories:
             duration = None
             # reset  with every new call as we need to set all arguments, such as tau, delay, again.