meet error running experiment with distracting background

[HaochenZhang717]

Dear authors,

thanks for your excellent work. I am trying to run your code but meet an error. I install a virtual environment as you said in readme file. But I meet the error below, do you know how to address it.

raceback (most recent call last):
File "/data/hchen/code/HRSSM/dreamer.py", line 449, in
main(parser.parse_args(remaining))
File "/data/hchen/code/HRSSM/dreamer.py", line 341, in main
state = tools.simulate(
File "/data/hchen/code/HRSSM/tools.py", line 272, in simulate
results = [r() for r in results]
File "/data/hchen/code/HRSSM/tools.py", line 272, in
results = [r() for r in results]
File "/data/hchen/code/HRSSM/parallel.py", line 206, in
return lambda: self._env.step(action)
File "/data/hchen/code/HRSSM/envs/wrappers.py", line 150, in step
return self._env.step({**action, self._key: clipped})
File "/data/hchen/code/HRSSM/envs/wrappers.py", line 120, in step
return self._env.step(action[self._key])
File "/data/hchen/code/HRSSM/envs/wrappers.py", line 18, in step
obs, reward, done, info = self._env.step(action)
File "/data/hchen/code/HRSSM/envs/wrappers.py", line 163, in step
time_step, reward, done, _, info = self._env.step(action)
File "/home/hchen/anaconda3/envs/hrssm/lib/python3.9/site-packages/gym/wrappers/time_limit.py", line 50, in step
observation, reward, terminated, truncated, info = self.env.step(action)
File "/home/hchen/anaconda3/envs/hrssm/lib/python3.9/site-packages/gym/wrappers/order_enforcing.py", line 37, in step
return self.env.step(action)
File "/home/hchen/anaconda3/envs/hrssm/lib/python3.9/site-packages/gym/wrappers/env_checker.py", line 37, in step
return env_step_passive_checker(self.env, action)
File "/home/hchen/anaconda3/envs/hrssm/lib/python3.9/site-packages/gym/utils/passive_env_checker.py", line 246, in env_step_passive_checker
check_obs(obs, env.observation_space, "step")
File "/home/hchen/anaconda3/envs/hrssm/lib/python3.9/site-packages/gym/utils/passive_env_checker.py", line 156, in check_obs
assert isinstance(obs, dict), f"{pre} must be a dict, actual type: {type(obs)}"
AssertionError: The obs returned by the step() method must be a dict, actual type: <class 'dm_env._environment.TimeStep'>