Finish migration of Dopamine to using discrete_domains instead of atari.

PiperOrigin-RevId: 231432169
ribeirojose · Jan 31, 2019 · 034e0ab · 034e0ab
1 parent 206c57d
commit 034e0ab
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 7 deletions.
diff --git a/README.md b/README.md
@@ -147,11 +147,11 @@ python tests/atari_init_test.py
 ```
 
 The entry point to the standard Atari 2600 experiment is
-[`dopamine/atari/train.py`](https://github.com/google/dopamine/blob/master/dopamine/atari/train.py).
+[`dopamine/discrete_domains/train.py`](https://github.com/google/dopamine/blob/master/dopamine/discrete_domains/train.py).
 To run the basic DQN agent,
 
 ```
-python -um dopamine.atari.train \
+python -um dopamine.discrete_domains.train \
   --agent_name=dqn \
   --base_dir=/tmp/dopamine \
   --gin_files='dopamine/agents/dqn/configs/dqn.gin'

diff --git a/dopamine/agents/rainbow/configs/c51_icml.gin b/dopamine/agents/rainbow/configs/c51_icml.gin
@@ -1,5 +1,4 @@
 # Hyperparameters used in Bellemare et al. (2017).
-import dopamine.atari.preprocessing
 import dopamine.agents.rainbow.rainbow_agent
 import dopamine.discrete_domains.atari_lib
 import dopamine.discrete_domains.run_experiment

diff --git a/dopamine/colab/agents.ipynb b/dopamine/colab/agents.ipynb
@@ -76,7 +76,7 @@
         "import numpy as np\n",
         "import os\n",
         "from dopamine.agents.dqn import dqn_agent\n",
-        "from dopamine.atari import run_experiment\n",
+        "from dopamine.discrete_domains import run_experiment\n",
         "from dopamine.colab import utils as colab_utils\n",
         "from absl import flags\n",
         "\n",
@@ -411,4 +411,4 @@
       ]
     }
   ]
-}
+}
diff --git a/dopamine/colab/load_statistics.ipynb b/dopamine/colab/load_statistics.ipynb
@@ -79,7 +79,7 @@
         "import numpy as np\n",
         "import os\n",
         "from dopamine.agents.dqn import dqn_agent\n",
-        "from dopamine.atari import run_experiment\n",
+        "from dopamine.discrete_domains import run_experiment\n",
         "from dopamine.colab import utils as colab_utils\n",
         "from absl import flags\n",
         "\n",
@@ -310,4 +310,4 @@
       ]
     }
   ]
-}
+}