doc change

mosaicml · Nov 2, 2024 · 827e7e5 · 827e7e5
1 parent 537a5ae
commit 827e7e5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/streaming/base/util.py b/streaming/base/util.py
@@ -491,6 +491,7 @@ def retry(  # type: ignore
 
             def clean_up():
                 # Do clean up stuff here
+                print("cleaning up")
 
             @retry(RuntimeError, clean_up_fn=clean_up, num_attempts=3, initial_backoff=0.1)
             def flaky_function():