fix: ant normalize reward

typoverflow · Aug 4, 2023 · 93bfa58 · 93bfa58
1 parent 93df49e
commit 93bfa58
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/offlinerllib/__init__.py b/offlinerllib/__init__.py
@@ -1,2 +1,2 @@
 
-__version__ = "0.0.12"
+__version__ = "0.0.13"
diff --git a/offlinerllib/utils/d4rl.py b/offlinerllib/utils/d4rl.py
@@ -126,7 +126,7 @@ def get_d4rl_dataset(task, normalize_reward=False, normalize_obs=False, terminat
     if normalize_reward:
         if "antmaze" in task:
             dataset, _ = antmaze_normalize_reward(dataset)
-        elif "halfcheetah" in task or "hopper" in task or "walker2d" in task:
+        elif "halfcheetah" in task or "hopper" in task or "walker2d" in task or "ant" in task:
             dataset, _ = mujoco_normalize_reward(dataset)
     termination_fn = get_termination_fn(task)
     if normalize_obs: