amend

pytorch · Jan 22, 2024 · 5961bd9 · 5961bd9
1 parent f71ccbb
commit 5961bd9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/torchrl/data/datasets/atari_dqn.py b/torchrl/data/datasets/atari_dqn.py
@@ -386,8 +386,9 @@ def available_datasets(cls):
         ]
         return ["/".join((game, str(loop))) for game in games for loop in range(1, 6)]
 
-    tmpdir = "/Users/vmoens/.cache/atari_root"
 
+    # If we want to keep track of the original atari files
+    tmpdir = None
     # use _max_runs for debugging, avoids downloading the entire dataset
     _max_runs = None