skip log (#3446)

mosaicml · Jul 1, 2024 · 8fd9c11 · 8fd9c11
1 parent 29ed31e
commit 8fd9c11
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/composer/loggers/mosaicml_logger.py b/composer/loggers/mosaicml_logger.py
@@ -146,7 +146,8 @@ def predict_end(self, state: State, logger: Logger) -> None:
         self._flush_metadata(force_flush=True)
 
     def close(self, state: State, logger: Logger) -> None:
-        self._flush_metadata(force_flush=True, future=False)
+        # Skip flushing metadata as it should be logged by fit/eval/predict_end. Flushing here
+        # might schedule futures while interpreter is shutting down, which will raise an error.
         if self._enabled:
             wait(self._futures)  # Ignore raised errors on close