From 2435cde7549ec834a913af2842ddbe624c51a47c Mon Sep 17 00:00:00 2001 From: Federico Pigozzi Date: Sun, 24 Sep 2023 18:26:26 +0200 Subject: [PATCH 1/2] obsolete legacy_spring parameter --- evojax/task/brax_task.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/evojax/task/brax_task.py b/evojax/task/brax_task.py index 9ae88f65..6255edb4 100644 --- a/evojax/task/brax_task.py +++ b/evojax/task/brax_task.py @@ -69,8 +69,7 @@ def __init__(self, self.test = test brax_env = create( env_name=env_name, - episode_length=max_steps, - legacy_spring=legacy_spring, + episode_length=max_steps ) self.obs_shape = tuple([brax_env.observation_size, ]) self.act_shape = tuple([brax_env.action_size, ]) From 132abe9bae8cd331f8e5d646f692783eeaf8123b Mon Sep 17 00:00:00 2001 From: Federico Pigozzi Date: Sun, 24 Sep 2023 18:35:02 +0200 Subject: [PATCH 2/2] Fix also constructor --- evojax/task/brax_task.py | 1 - 1 file changed, 1 deletion(-) diff --git a/evojax/task/brax_task.py b/evojax/task/brax_task.py index 6255edb4..8ee21834 100644 --- a/evojax/task/brax_task.py +++ b/evojax/task/brax_task.py @@ -61,7 +61,6 @@ class BraxTask(VectorizedTask): def __init__(self, env_name: str, max_steps: int = 1000, - legacy_spring: bool = True, bd_extractor: Optional[BDExtractor] = None, test: bool = False): self.max_steps = max_steps