Update

[ghstack-poisoned]
pytorch · Dec 16, 2024 · c29aef6 · c29aef6
1 parent 1402fae
commit c29aef6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sota-implementations/td3/config.yaml b/sota-implementations/td3/config.yaml
@@ -21,7 +21,7 @@ collector:
 replay_buffer:
   prb: 0 # use prioritized experience replay
   size: 1000000
-  scratch_dir:
+  scratch_dir: null
 
 # optim
 optim:

diff --git a/sota-implementations/td3/utils.py b/sota-implementations/td3/utils.py
@@ -144,7 +144,7 @@ def make_replay_buffer(
         prefetch = 0
     with (
         tempfile.TemporaryDirectory()
-        if scratch_dir is None
+        if scratch_dir in ("", None)
         else nullcontext(scratch_dir)
     ) as scratch_dir:
         storage_cls = (