diff --git a/iblrig/base_tasks.py b/iblrig/base_tasks.py index 1ba71d104..6c86f7dc3 100644 --- a/iblrig/base_tasks.py +++ b/iblrig/base_tasks.py @@ -59,7 +59,7 @@ class HasBpod(Protocol): class BaseSession(ABC): version = None - """str: !!CURRENTLY UNUSED!! task version string.""" + """str: Task version string.""" # protocol_name: str | None = None """str: The name of the task protocol (NB: avoid spaces).""" base_parameters_file: Path | None = None @@ -431,6 +431,8 @@ def _make_task_parameters_dict(self): } with contextlib.suppress(importlib.metadata.PackageNotFoundError): patch_dict['PROJECT_EXTRACTION_VERSION'] = importlib.metadata.version('project_extraction') + if self.version is not None: + patch_dict['TASK_VERSION'] = self.version output_dict.update(patch_dict) return output_dict