Add pre-commit

Netflix · Jun 26, 2024 · 7919204 · 7919204
1 parent 4ebcbe2
commit 7919204
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git 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
@@ -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(