Does rollout.py work with qmix?

[nsa26]

I am trying to evaluate qmix trained model using rollout.py but I am getting the following:

checkpoint: /home/ray_results/QMIX2/checkpoint_2/checkpoint-2 2021-03-22 17:27:39,867 INFO trainable.py:489 -- Current state after restoring: {'_iteration': 2, '_timesteps_total': None, '_time_total': 4.807272911071777, '_episodes_total': 200} Traceback (most recent call last): File "rollout.py", line 562, in run(args, parser) File "rollout.py", line 376, in run True, None) File "rollout.py", line 509, in rollout policy_id=policy_id) File "/home/.local/lib/python3.6/site-packages/ray/rllib/agents/trainer.py", line 831, in compute_action explore=explore) File "/home/.local/lib/python3.6/site-packages/ray/rllib/policy/policy.py", line 207, in compute_single_action timestep=timestep) File "/home/.local/lib/python3.6/site-packages/ray/rllib/agents/qmix/qmix_policy.py", line 287, in compute_actions explore=explore) File "/home/.local/lib/python3.6/site-packages/ray/rllib/utils/exploration/epsilon_greedy.py", line 80, in get_exploration_action timestep) File "/home/.local/lib/python3.6/site-packages/ray/rllib/utils/exploration/epsilon_greedy.py", line 146, in _get_torch_exploration_action epsilon = self.epsilon_schedule(self.last_timestep) File "/home/.local/lib/python3.6/site-packages/ray/rllib/utils/schedules/schedule.py", line 45, in call return self.value(t) File "/home/.local/lib/python3.6/site-packages/ray/rllib/utils/schedules/schedule.py", line 40, in value return self._value(t) File "/home/.local/lib/python3.6/site-packages/ray/rllib/utils/schedules/piecewise_schedule.py", line 53, in _value if l_t <= t < r_t: TypeError: '<=' not supported between instances of 'int' and 'NoneType'