Optionally surface exceptions that happen under executors by reading

[python_utils.git] / state_tracker.py

diff --git a/state_tracker.py b/state_tracker.py
index 4836e3eb99915a6d25a06102ee0c93163cd69133..453faf7b1972d8f4f1b3250bfdd353150e682503 100644 (file)
--- a/state_tracker.py
+++ b/state_tracker.py
@@ -73,21 +73,15 @@ class StateTracker(ABC):
         for update_id in sorted(self.last_reminder_ts.keys()):
             if force_all_updates_to_run:
                 logger.debug('Forcing all updates to run')
-                self.update(
-                    update_id, self.now, self.last_reminder_ts[update_id]
-                )
+                self.update(update_id, self.now, self.last_reminder_ts[update_id])
                 self.last_reminder_ts[update_id] = self.now
                 return
 
             refresh_secs = self.update_ids_to_update_secs[update_id]
             last_run = self.last_reminder_ts[update_id]
             if last_run is None:  # Never run before
-                logger.debug(
-                    f'id {update_id} has never been run; running it now'
-                )
-                self.update(
-                    update_id, self.now, self.last_reminder_ts[update_id]
-                )
+                logger.debug(f'id {update_id} has never been run; running it now')
+                self.update(update_id, self.now, self.last_reminder_ts[update_id])
                 self.last_reminder_ts[update_id] = self.now
             else:
                 delta = self.now - last_run
@@ -148,9 +142,7 @@ class AutomaticStateTracker(StateTracker):
         This may block for as long as self.sleep_delay.
 
         """
-        logger.debug(
-            'Setting shutdown event and waiting for background thread.'
-        )
+        logger.debug('Setting shutdown event and waiting for background thread.')
         self.should_terminate.set()
         self.updater_thread.join()
         logger.debug('Background thread terminated.')