From 5961bd9c58d072ae6e02c77b2ebb6adcef9b3b39 Mon Sep 17 00:00:00 2001
From: vmoens <vincentmoens@gmail.com>
Date: Mon, 22 Jan 2024 16:26:52 +0000
Subject: [PATCH] amend

---
 torchrl/data/datasets/atari_dqn.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/torchrl/data/datasets/atari_dqn.py b/torchrl/data/datasets/atari_dqn.py
index cde6db8fe80..2e93d801783 100644
--- 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