Added more tests

2022-09-26 16:11:41 +02:00
parent ee4a46fad1
commit 6e4be06ffd
2 changed files with 105 additions and 1 deletions
@@ -141,7 +141,7 @@ def make_bb(
    Returns: DMP wrapped gym env

    """
-    _verify_time_limit(traj_gen_kwargs.get("duration", None), kwargs.get("time_limit", None))
+    _verify_time_limit(traj_gen_kwargs.get("duration"), kwargs.get("time_limit"))

    learn_sub_trajs = black_box_kwargs.get('learn_sub_trajectories')
    do_replanning = black_box_kwargs.get('replanning_schedule')