is it just this script that you are running that breaks? What happens if instead of pipe.upload_model
you callprint(pipe._get_pipeline_task())
?
Traceback (most recent call last): File "/home/mano/Desktop/askui-stuff/ml-models/pipelines/yolo/pipeline_yolo.py", line 163, in <module> print(pipe._get_pipeline_task()) File "/home/mano/anaconda3/envs/askuienv/lib/python3.9/site-packages/clearml/automation/controller.py", line 2649, in _get_pipeline_task if str(task.task_type) == str(Task.TaskTypes.controller) and cls._tag in task.get_system_tags(): AttributeError: 'NoneType' object has no attribute 'task_type'
That is very odd. Is the script above all you're running?
Don't call PipelineController
functions after start
has finished. Use a post_execute_callback
instead
` from clearml import PipelineController
def some_step():
return
def upload_model_to_controller(controller, node):
print("Start uploading the model")
if name == "main":
pipe = PipelineController(name="Yolo Pipeline Controller", project="yolo_pipelines", version="1.0.0")
pipe.add_function_step(
name="some_step",
function=some_step,
cache_executed_step=False,
post_execute_callback=upload_model_to_controller
)
pipe.start() `