updated saving nwb

talmolab · Aug 1, 2024 · 0a8fc45 · 0a8fc45
1 parent f88ca17
commit 0a8fc45
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sleap_io/io/main.py b/sleap_io/io/main.py
@@ -241,7 +241,7 @@ def save_file(
     elif format in ("nwb", "nwb_predictions"):
         save_nwb(labels, filename, as_training=False)
     elif format == "nwb_training":
-        save_nwb(labels, filename, as_training=True)
+        save_nwb(labels, filename, True, frame_inds=kwargs.pop("frame_inds", None))
     elif format == "labelstudio":
         save_labelstudio(labels, filename, **kwargs)
     elif format == "jabs":