diff --git a/laborious/workflows/predictions_batch.py b/laborious/workflows/predictions_batch.py index f79206f..afd7c86 100644 --- a/laborious/workflows/predictions_batch.py +++ b/laborious/workflows/predictions_batch.py @@ -114,4 +114,4 @@ class PredictionsBatch: } # Execute prediction process workflow - await workflow.execute_child_workflow('prediction_process', prediction_input) + await workflow.execute_child_workflow('subworkflow.prediction_process', prediction_input) diff --git a/laborious/workflows/sub_workflows/format_and_export_prediction.py b/laborious/workflows/sub_workflows/format_and_export_prediction.py index 12898ce..8d8c9da 100644 --- a/laborious/workflows/sub_workflows/format_and_export_prediction.py +++ b/laborious/workflows/sub_workflows/format_and_export_prediction.py @@ -10,7 +10,7 @@ with workflow.unsafe.imports_passed_through(): from laborious.activities.activities import Activities -@workflow.defn(name='format_and_export_prediction') +@workflow.defn(name='subworkflow.format_and_export_prediction') class FormatAndExportPrediction: """ Data formatting and export workflow for prediction results. diff --git a/laborious/workflows/sub_workflows/prediction_process.py b/laborious/workflows/sub_workflows/prediction_process.py index 7bd36e2..ab78fb1 100644 --- a/laborious/workflows/sub_workflows/prediction_process.py +++ b/laborious/workflows/sub_workflows/prediction_process.py @@ -9,7 +9,7 @@ with workflow.unsafe.imports_passed_through(): from laborious.activities.activities import Activities -@workflow.defn(name='prediction_process') +@workflow.defn(name='subworkflow.prediction_process') class PredictionProcess: """ Core prediction processing workflow for the Laborious system. @@ -194,7 +194,7 @@ class PredictionProcess: # Delegate to export workflow for data persistence await workflow.execute_child_workflow( - 'format_and_export_prediction', + 'subworkflow.format_and_export_prediction', { 'metadata': metadata, 'path_flag': path_flag, @@ -275,7 +275,7 @@ class PredictionProcess: elif path_flag == 'CONTINUE': # call write workflow await workflow.execute_child_workflow( - 'format_and_export_prediction', + 'subworkflow.format_and_export_prediction', { 'metadata': metadata, 'path_flag': path_flag, diff --git a/tests/laborious/workflows/subworkflows/test_prediction_process.py b/tests/laborious/workflows/subworkflows/test_prediction_process.py index 6baec30..0d8a0c1 100644 --- a/tests/laborious/workflows/subworkflows/test_prediction_process.py +++ b/tests/laborious/workflows/subworkflows/test_prediction_process.py @@ -170,7 +170,7 @@ async def test_run(workflow_mock, prediction_process): ) workflow_mock.execute_child_workflow.assert_called_once_with( - 'format_and_export_prediction', + 'subworkflow.format_and_export_prediction', { 'metadata': metadata, 'path_flag': 'continue', @@ -730,7 +730,7 @@ async def test_path_flag_handler_continue(workflow_mock, prediction_process): assert result is True workflow_mock.execute_activity_method.assert_not_called() workflow_mock.execute_child_workflow.assert_called_once_with( - 'format_and_export_prediction', + 'subworkflow.format_and_export_prediction', { 'metadata': metadata, 'path_flag': path_flag, diff --git a/tests/laborious/workflows/test_predictions_batch.py b/tests/laborious/workflows/test_predictions_batch.py index 62cc43c..7979c47 100644 --- a/tests/laborious/workflows/test_predictions_batch.py +++ b/tests/laborious/workflows/test_predictions_batch.py @@ -75,5 +75,5 @@ async def test_run(workflow_mock: AsyncMock, predictions_batch: PredictionsBatch } workflow_mock.execute_child_workflow.assert_has_calls( - [call('prediction_process', prediction_input)] + [call('subworkflow.prediction_process', prediction_input)] )