From 5e1d26daa4778405317ffcf9d46813203684894f Mon Sep 17 00:00:00 2001
From: Sam Bray <sam.bray@ucsf.edu>
Date: Tue, 26 Nov 2024 12:01:17 -0800
Subject: [PATCH] generalize DecodingParameters fetch1

---
 src/spyglass/decoding/v1/core.py | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/src/spyglass/decoding/v1/core.py b/src/spyglass/decoding/v1/core.py
index f43f0948b..453741291 100644
--- a/src/spyglass/decoding/v1/core.py
+++ b/src/spyglass/decoding/v1/core.py
@@ -97,7 +97,20 @@ def fetch(self, *args, **kwargs):
     def fetch1(self, *args, **kwargs):
         """Return one decoding paramset as a class."""
         row = super().fetch1(*args, **kwargs)
-        row["decoding_params"] = restore_classes(row["decoding_params"])
+
+        if len(args) == 0:
+            row["decoding_params"] = restore_classes(row["decoding_params"])
+            return row
+
+        if "decoding_params" in args:
+            if len(args) == 1:
+                return restore_classes(row)
+            row = list(row)
+            row[args.index("decoding_params")] = restore_classes(
+                row[args.index("decoding_params")]
+            )
+            return tuple(row)
+
         return row