ImportError: cannot import name 'ObservationInput'

[sweetice]

Following readme.md
When I run

python -m qmap.train_mario --render

The result is:

Traceback (most recent call last):
  File "/home/mirror/anaconda3/envs/qmap/lib/python3.6/runpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
  File "/home/mirror/anaconda3/envs/qmap/lib/python3.6/runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "/home/mirror/Documents/GithubCode/qmap/qmap/train_mario.py", line 9, in <module>
    from qmap.agents.q_map_dqn_agent import Q_Map_DQN_Agent
  File "/home/mirror/Documents/GithubCode/qmap/qmap/agents/q_map_dqn_agent.py", line 3, in <module>
    from baselines.deepq.utils import ObservationInput
ImportError: cannot import name 'ObservationInput'

baselines version is:

(qmap) mirror@agent:~/Desktop$ pip list | grep base
baselines           0.1.5