From 725e6c8fe17a1b9a5dcec42a7abebf25c587f82e Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Mon, 18 Sep 2023 19:18:26 +0200
Subject: [PATCH] Fixed wrong env name (still contained mp info)

---
 fancy_gym/dmc/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/fancy_gym/dmc/__init__.py b/fancy_gym/dmc/__init__.py
index 6e9f38c..7fcebba 100644
--- a/fancy_gym/dmc/__init__.py
+++ b/fancy_gym/dmc/__init__.py
@@ -32,7 +32,7 @@ register(
 _dmc_cartpole_tasks = ["balance", "balance_sparse", "swingup", "swingup_sparse"]
 for _task in _dmc_cartpole_tasks:
     register(
-        id=f'dm_control/cartpole-{_task}_dmp-v0',
+        id=f'dm_control/cartpole-{_task}-v0',
         register_step_based=False,
         mp_wrapper=suite.cartpole.MPWrapper,
         add_mp_types=['DMP', 'ProMP'],