Add assertion to aid type checking.

PiperOrigin-RevId: 568244870 Change-Id: I66cac69883a9857a3177a8710249e66987683941
google-deepmind · Sep 25, 2023 · 32009e2 · 32009e2
1 parent 74a841a
commit 32009e2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/acme/utils/loggers/auto_close.py b/acme/utils/loggers/auto_close.py
@@ -35,6 +35,8 @@ def __init__(self, logger: base.Logger):
     self._finalizer = weakref.finalize(self, logger.close)
 
   def write(self, values: base.LoggingData):
+    if self._logger is None:
+      raise ValueError('init not called')
     self._logger.write(values)
 
   def close(self):