From 75c138b23bc8e4970e63305b882afe97d18b8b68 Mon Sep 17 00:00:00 2001
From: Vincent Moens <vmoens@meta.com>
Date: Sun, 15 Dec 2024 19:41:48 -0800
Subject: [PATCH] Update (base update)

[ghstack-poisoned]
---
 sota-implementations/td3/config.yaml | 2 +-
 sota-implementations/td3/utils.py    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/sota-implementations/td3/config.yaml b/sota-implementations/td3/config.yaml
index ba2db81b489..8207a41a9f5 100644
--- 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
index 8395d9eeaec..071bad6c68b 100644
--- 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 = (