Internal change.

PiperOrigin-RevId: 553134800 Change-Id: I9880dbd0ee617d37bb884cedabe591f44120e58b
google-deepmind · Aug 2, 2023 · 95b7cfe · 95b7cfe
1 parent fb87fd5
commit 95b7cfe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/acme/adders/reverb/structured.py b/acme/adders/reverb/structured.py
@@ -153,7 +153,7 @@ def add(self,
           extras: types.NestedArray = ()):
     """Record an action and the following timestep."""
 
-    if not self._writer.step_is_open:
+    if self._writer is None or not self._writer.step_is_open:
       raise ValueError('adder.add_first must be called before adder.add.')
 
     # Add the timestep to the buffer.