-
Notifications
You must be signed in to change notification settings - Fork 57
Description
Describe the bug
If your Python class name for an Artifact Task (e.g. NumpyArtifactTask) doesn't match the registered name (i.e. you name it NPArtifactTask instead), then the job hangs forever in the "Started" state even though the jobs show a SystemExit / Fail.
Worker logs:
26-02-23T19:08:37.678997Z [info ] tensorflow-cpu: dioptra.rq.tasks.run_v1_dioptra_job(experiment_id=37, job_id=38) (38) [rq.worker]
2026-02-23T19:08:39.490104Z [error ] Failed to locate artifact task: PngBytesArtifactTask [dioptra.rq.tasks.run_v1_dioptra_job] experiment_id=37 job_id=38
2026-02-23T19:08:39.524437Z [error ] Worker 96a17528936f458fa24409f118ab96ba: job 38: exception raised while executing (dioptra.rq.tasks.run_v1_dioptra_job)
Traceback (most recent call last):
File "/opt/venv/lib/python3.11/site-packages/rq/worker.py", line 1439, in perform_job
return_value = job.perform()
^^^^^^^^^^^^^
File "/opt/venv/lib/python3.11/site-packages/rq/job.py", line 1318, in perform
self._result = self._execute()
^^^^^^^^^^^^^^^
File "/opt/venv/lib/python3.11/site-packages/rq/job.py", line 1376, in _execute
result = self.func(*self.args, **self.kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/venv/lib/python3.11/site-packages/dioptra/rq/tasks/run_v1_dioptra_job.py", line 44, in run_v1_dioptra_job
run_dioptra_job.main(
File "/opt/venv/lib/python3.11/site-packages/dioptra/sdk/utilities/run_dioptra_job.py", line 184, in main
_create_engine_schema(context=context, plugins=artifact_plugins, log=log)
File "/opt/venv/lib/python3.11/site-packages/dioptra/sdk/utilities/run_dioptra_job.py", line 507, in _create_engine_schema
exit(1)
File "", line 26, in call
SystemExit: 1
To Reproduce
Steps to reproduce the behavior:
- Create Python Artifact Class with name
X - Register it with a different name
Y - Create an entrypoint that uses that class
- Try and execute a job
Expected behavior
Job should fail and error message should be logged
Desktop (please complete the following information):
- OS: WSL2
- Browser Chrome
Server
- OS: Ubuntu
- Virtualization: Docker