Skip to content

Commit

Permalink
Bugfix to gym version comparison, now uses package "packaging" for this.
Browse files Browse the repository at this point in the history
  • Loading branch information
jfrancis71 committed Sep 21, 2022
1 parent c7cc2df commit f09e108
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion pfrl/wrappers/atari_wrappers.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
import gym
import numpy as np
from gym import spaces
from packaging import version

import pfrl

Expand Down Expand Up @@ -37,7 +38,7 @@ def reset(self, **kwargs):
if self.override_num_noops is not None:
noops = self.override_num_noops
else:
if gym.__version__ >= "0.24.0":
if version.parse(gym.__version__) >= version.parse("0.24.0"):
noops = self.unwrapped.np_random.integers(1, self.noop_max + 1)
else:
noops = self.unwrapped.np_random.randint(1, self.noop_max + 1)
Expand Down

0 comments on commit f09e108

Please sign in to comment.