From 025967014e7a987c331da4d5ad30bbd6cb85c7ee Mon Sep 17 00:00:00 2001 From: Sam Bray Date: Wed, 10 Jan 2024 12:52:43 -0800 Subject: [PATCH] restrict get position info by session name --- src/spyglass/decoding/v1/clusterless.py | 5 ++++- src/spyglass/decoding/v1/sorted_spikes.py | 5 ++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/src/spyglass/decoding/v1/clusterless.py b/src/spyglass/decoding/v1/clusterless.py index 79367f793..3d0e36aa3 100644 --- a/src/spyglass/decoding/v1/clusterless.py +++ b/src/spyglass/decoding/v1/clusterless.py @@ -305,7 +305,10 @@ def load_environments(key): @staticmethod def load_position_info(key): - position_group_key = {"position_group_name": key["position_group_name"]} + position_group_key = { + "position_group_name": key["position_group_name"], + "nwb_file_name": key["nwb_file_name"], + } position_variable_names = (PositionGroup & position_group_key).fetch1( "position_variables" ) diff --git a/src/spyglass/decoding/v1/sorted_spikes.py b/src/spyglass/decoding/v1/sorted_spikes.py index a00985870..d8c934c68 100644 --- a/src/spyglass/decoding/v1/sorted_spikes.py +++ b/src/spyglass/decoding/v1/sorted_spikes.py @@ -296,7 +296,10 @@ def load_environments(key): @staticmethod def load_position_info(key): - position_group_key = {"position_group_name": key["position_group_name"]} + position_group_key = { + "position_group_name": key["position_group_name"], + "nwb_file_name": key["nwb_file_name"], + } position_variable_names = (PositionGroup & position_group_key).fetch1( "position_variables" )