Skip to content

Commit

Permalink
Fix gym->gymnasium import in SQIL implementation.
Browse files Browse the repository at this point in the history
  • Loading branch information
ernestum committed Aug 29, 2023
1 parent 7ad6d8e commit ed00a6f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/imitation/algorithms/sqil.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit ed00a6f

Please sign in to comment.