lightning-pose-app 1.8.1a18__tar.gz → 1.8.1a19__tar.gz

{lightning_pose_app-1.8.1a18 → lightning_pose_app-1.8.1a19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lightning-pose-app
-Version: 1.8.1a18
+Version: 1.8.1a19
 Summary: 
 Requires-Python: >=3.10
 Classifier: Programming Language :: Python :: 3

{lightning_pose_app-1.8.1a18 → lightning_pose_app-1.8.1a19}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [project]
 name = "lightning-pose-app"
-version = "1.8.1a18"
+version = "1.8.1a19"
 requires-python = ">=3.10"
 dependencies = [
     "aiofiles~=24.1.0",

{lightning_pose_app-1.8.1a18 → lightning_pose_app-1.8.1a19}/src/litpose_app/routes/labeler/bundle_adjust.py

@@ -146,14 +146,19 @@ def _bundle_adjust_impl(
                 f"Insufficient frames found after filtering for session {request.sessionKey}. Possible error in session extraction logic.")
 
     # Remove rows with NaN coordinates
+    # Get indices of rows with NaN coordinates in any view
+    nan_indices = set()
     for view in views:
         df = dfs_by_view[view]
         picked_columns = [c for c in df.columns if c[2] in ("x", "y")]
-        dfs_by_view[view] = df.loc[:, picked_columns].dropna()
+        nan_indices.update(df.loc[:, picked_columns].dropna().index.symmetric_difference(df.index))
+
+    # Drop those indices from all views
+    for view in views:
+        dfs_by_view[view] = dfs_by_view[view].drop(nan_indices)
         if len(dfs_by_view[view]) == 0:
             raise RuntimeError(
                 f"Insufficient frames found after dropping NaN rows for session {request.sessionKey}.")
-
     # Normalize columns: x, y alternating.
     for view in views:
         df = dfs_by_view[view]