From ed00a6f208f1bdb9fd86fc84187276492ad7346a Mon Sep 17 00:00:00 2001 From: Maximilian Ernestus Date: Tue, 29 Aug 2023 17:37:56 +0200 Subject: [PATCH] Fix gym->gymnasium import in SQIL implementation. --- src/imitation/algorithms/sqil.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/imitation/algorithms/sqil.py b/src/imitation/algorithms/sqil.py index 85c71d890..80f1a897d 100644 --- a/src/imitation/algorithms/sqil.py +++ b/src/imitation/algorithms/sqil.py @@ -8,7 +8,7 @@ import numpy as np import torch as th -from gym import spaces +from gymnasium import spaces from stable_baselines3 import dqn from stable_baselines3.common import buffers, policies, type_aliases, vec_env from stable_baselines3.dqn.policies import DQNPolicy