From 791920499553774e0de5a27205ee7b92704fc476 Mon Sep 17 00:00:00 2001
From: Shashank Srikanth <ssrikanth@netflix.com>
Date: Tue, 25 Jun 2024 18:08:53 -0700
Subject: [PATCH] Add pre-commit

---
 metaflow/system/system_monitor.py | 14 ++++++++++----
 metaflow/task.py                  |  2 +-
 2 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/metaflow/system/system_monitor.py b/metaflow/system/system_monitor.py
index 0e174d2c10..b98b73992f 100644
--- a/metaflow/system/system_monitor.py
+++ b/metaflow/system/system_monitor.py
@@ -26,11 +26,17 @@ def update_context(self, context: Dict[str, Any]):
 
         """
         from metaflow.sidecar import Message, MessageTypes
+
         self._context.update(context)
-        self.monitor.send(Message(MessageTypes.MUST_SEND, {
-            "is_context_updated": True,
-            **self._context,
-        }))
+        self.monitor.send(
+            Message(
+                MessageTypes.MUST_SEND,
+                {
+                    "is_context_updated": True,
+                    **self._context,
+                },
+            )
+        )
 
     def init_system_monitor(
         self, flow_name: str, monitor: "metaflow.monitor.NullMonitor"
diff --git a/metaflow/task.py b/metaflow/task.py
index abcbd3ec76..e74302e2f2 100644
--- a/metaflow/task.py
+++ b/metaflow/task.py
@@ -317,7 +317,7 @@ def clone_only(
                 level="info",
                 module="metaflow.task",
                 name="clone",
-                payload={**task_payload,"msg": msg},
+                payload={**task_payload, "msg": msg},
             )
         # If we actually have to do the clone ourselves, proceed...
         clone_task_helper(