From 0a8fc45b2dccb2560fcc5e113c636935db70d317 Mon Sep 17 00:00:00 2001 From: Keya Loding Date: Thu, 1 Aug 2024 15:08:01 -0700 Subject: [PATCH] updated saving nwb --- sleap_io/io/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sleap_io/io/main.py b/sleap_io/io/main.py index 65824e51..6c3d8f92 100644 --- 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":