diff --git a/d3rlpy/cli.py b/d3rlpy/cli.py index 2b560ff5..72833e47 100644 --- a/d3rlpy/cli.py +++ b/d3rlpy/cli.py @@ -393,7 +393,7 @@ def print_available_options() -> None: ["minari[all]>=0.5.1", "gymnasium-robotics>=1.2.4"], upgrade=True ) elif name == "dm_control": - _install_module(["shimmy[dm-control]==1.3.0"], upgrade=True) + _install_module(["shimmy[dm-control]==2.0.0"], upgrade=True) elif name == "list": print_available_options() else: