fix bug in ShelfWriter: top-level index required to write metadata to shelve

deruyter92 · deruyter92 · commit fe00c00a8369 · 2026-02-19T16:22:10.000+01:00
diff --git a/deeplabcut/pose_estimation_pytorch/runners/shelving.py b/deeplabcut/pose_estimation_pytorch/runners/shelving.py
@@ -149,9 +149,12 @@ def add_prediction(
         self._frame_index += 1
 
     def close(self) -> None:
-        """Opens the shelf"""
+        """Closes the shelf and writes the updated metadata"""
         if self._open and self._frame_index > 0:
-            self._db["metadata"]["nframes"] = self._frame_index
+            # Write updated metadata to shelf (top-level indexing required for shelve)
+            metadata = self._db["metadata"]
+            metadata["nframes"] = self._frame_index
+            self._db["metadata"] = metadata
 
         super().close()
 
