diff --git a/src/dynamic_foraging_processing/pipeline/_pipeline.py b/src/dynamic_foraging_processing/pipeline/_pipeline.py index 0796cdc..f7a3842 100644 --- a/src/dynamic_foraging_processing/pipeline/_pipeline.py +++ b/src/dynamic_foraging_processing/pipeline/_pipeline.py @@ -279,8 +279,16 @@ def _write_processing( experimenters=["Alex Piet", "Micah Woodard", "Bruno Cruz", "Arjun Sridhar"], start_date_time=start_date_time, end_date_time=end_date_time, + pipeline_name="dynamic-foraging-processing-pipeline", ) - ] + ], + pipelines=[ + Code( + url=_CODE_URL, + version=_PACKAGE_VERSION, + name="dynamic-foraging-processing-pipeline", + ) + ], ) processing.write_standard_file(output_directory=Path(output_path)) diff --git a/tests/test_pipeline/test_pipeline.py b/tests/test_pipeline/test_pipeline.py index cef052f..7af0cbf 100644 --- a/tests/test_pipeline/test_pipeline.py +++ b/tests/test_pipeline/test_pipeline.py @@ -371,6 +371,11 @@ def test_run_nwb_writes_nwb_and_processing(tmp_path): assert processing_json.exists() loaded = Processing.model_validate_json(processing_json.read_text()) assert loaded.data_processes[0].process_type == ProcessName.PIPELINE + # The data process and the top-level pipeline are linked by name. + assert loaded.data_processes[0].pipeline_name == "dynamic-foraging-processing-pipeline" + assert [p.name for p in loaded.pipelines] == ["dynamic-foraging-processing-pipeline"] + assert loaded.pipelines[0].version == _pipeline._PACKAGE_VERSION + assert loaded.pipelines[0].url == _pipeline._CODE_URL def test_run_qc_writes_quality_control(tmp_path):