From e2b41828af969f2b0c6450a53b8827946ff8673d Mon Sep 17 00:00:00 2001
From: Mathieu Poliquin <mathieu.poliquin@gmail.com>
Date: Tue, 30 Oct 2018 04:30:41 +0800
Subject: [PATCH] Set 'cnn' as default network for retro (#683)

---
 baselines/run.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/baselines/run.py b/baselines/run.py
index 4aaf1a7..28cf620 100644
--- a/baselines/run.py
+++ b/baselines/run.py
@@ -131,7 +131,7 @@ def get_env_type(env_id):
 
 
 def get_default_network(env_type):
-    if env_type == 'atari':
+    if env_type in {'atari', 'retro'}:
         return 'cnn'
     else:
         return 'mlp'