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(