ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 13:23:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix missing component callbacks on multiple step calls (#1195)

Browse Source
This commit is contained in:
Matteo Destro 2024-02-01 15:04:38 +01:00 committed by GitHub
parent 831c1394da
commit 56ba25f554
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
clearml/automation/controller.py
View File

@ -4075,6 +4075,13 @@ class PipelineDecorator(PipelineController):
): ):
_node.name = "{}_{}".format(_node_name, counter) _node.name = "{}_{}".format(_node_name, counter)
counter += 1 counter += 1
# Copy callbacks to the replicated node
if cls._singleton._pre_step_callbacks.get(_node_name):
cls._singleton._pre_step_callbacks[_node.name] = cls._singleton._pre_step_callbacks[_node_name]
if cls._singleton._post_step_callbacks.get(_node_name):
cls._singleton._post_step_callbacks[_node.name] = cls._singleton._post_step_callbacks[_node_name]
if cls._singleton._status_change_callbacks.get(_node_name):
cls._singleton._status_change_callbacks[_node.name] = cls._singleton._status_change_callbacks[_node_name]
_node_name = _node.name _node_name = _node.name
if _node.name not in cls._singleton._nodes: if _node.name not in cls._singleton._nodes:
cls._singleton._nodes[_node.name] = _node cls._singleton._nodes[_node.name] = _node