From 5aee1eb063480cd16dc5b88bd2296fa94c4ee4d4 Mon Sep 17 00:00:00 2001 From: Miles Wells Date: Thu, 6 Mar 2025 17:45:13 +0200 Subject: [PATCH] PREGENERATED_SESSION_NUM -> SESSION_TEMPLATE_ID in ephys passive extraction --- ibllib/io/extractors/ephys_passive.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/ibllib/io/extractors/ephys_passive.py b/ibllib/io/extractors/ephys_passive.py index c6af7c813..e41550010 100644 --- a/ibllib/io/extractors/ephys_passive.py +++ b/ibllib/io/extractors/ephys_passive.py @@ -68,9 +68,11 @@ def _load_passive_session_fixtures(session_path: str, task_collection: str = 'ra :rtype: dict """ - # THIS CAN BE PREGENERATED SESSION NO + # The pregenerated session number has had many parameter names settings = rawio.load_settings(session_path, task_collection=task_collection) - ses_nb = settings['PREGENERATED_SESSION_NUM'] + pars = map(settings.get, ['PRELOADED_SESSION_NUM', 'PREGENERATED_SESSION_NUM', 'SESSION_TEMPLATE_ID']) + ses_nb = next((k for k in pars if k is not None), None) + session_order = settings.get('SESSION_ORDER', None) if session_order: # TODO test this out and make sure it okay assert settings["SESSION_ORDER"][settings["SESSION_IDX"]] == ses_nb