From 1402faeadab945dc4f5f1106c8ee70d77aa68750 Mon Sep 17 00:00:00 2001
From: Vincent Moens <vmoens@meta.com>
Date: Sun, 15 Dec 2024 19:39:37 -0800
Subject: [PATCH] Update

[ghstack-poisoned]
---
 sota-implementations/td3/utils.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/sota-implementations/td3/utils.py b/sota-implementations/td3/utils.py
index 3f797064f35..8395d9eeaec 100644
--- a/sota-implementations/td3/utils.py
+++ b/sota-implementations/td3/utils.py
@@ -140,6 +140,8 @@ def make_replay_buffer(
     prefetch: int = 3,
     compile: bool = False,
 ):
+    if compile:
+        prefetch = 0
     with (
         tempfile.TemporaryDirectory()
         if scratch_dir is None