lightning-pose-app 1.8.1a18__py3-none-any.whl → 1.8.1a19__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: lightning-pose-app
3
- Version: 1.8.1a18
3
+ Version: 1.8.1a19
4
4
  Summary:
5
5
  Requires-Python: >=3.10
6
6
  Classifier: Programming Language :: Python :: 3
@@ -32,7 +32,7 @@ litpose_app/routes/configs.py,sha256=k2Zok69l7CqEoprVQxqV5YHjQucaC7w6oS2qoKpx0UU
32
32
  litpose_app/routes/extract_frames.py,sha256=HFh6oC8yJg6fa-Ob5CTIsqMgReGn6vdrqLe_5HRYM0U,4084
33
33
  litpose_app/routes/ffprobe.py,sha256=UVKDu4ghXkYNuzI-91KPMNpQukHV9Goeps7ex_lG_SU,5526
34
34
  litpose_app/routes/labeler/__init__.py,sha256=HyXqJ9xNdXA2jEoKrpy3-5AZrob7NbaedzyIqkmLBZA,1387
35
- litpose_app/routes/labeler/bundle_adjust.py,sha256=axIKqPGbOdDlxFBdR75XF4R5NWOd63QLRMAJ5eb3ATE,8049
35
+ litpose_app/routes/labeler/bundle_adjust.py,sha256=dRLPh9zb4tvqVU87mP4SWPx00D7dmzx34s2c0hznN3g,8296
36
36
  litpose_app/routes/labeler/find_label_files.py,sha256=6jZeJ7lemAaRNukLMSu74kGQ3W9c80Q7BzQHDG42GBk,5060
37
37
  litpose_app/routes/labeler/multiview_autolabel.py,sha256=BpAH_tySDfn8wK6C-H8iLnZm4NW4DAM6NBXMwAp0e2E,5434
38
38
  litpose_app/routes/labeler/save_mvframe.py,sha256=toE5xW93SqQmWNFHFASrtI2Akqm_ghDNigXYngMS7KM,6248
@@ -57,6 +57,6 @@ litpose_app/utils/mv_label_file.py,sha256=VarVDSF0Rx90WThLZfArS7nmKiGcTmONNWjOGl
57
57
  litpose_app/utils/video/__init__.py,sha256=dSc-yVjXmn7ajNR7WgkCUUFdn9RLS2NBM2CjH0ePQXE,464
58
58
  litpose_app/utils/video/export_frames.py,sha256=kAIqjMarqTe51mJd3jk9xw4MotEUV1UfxtgTepyw2mc,1738
59
59
  litpose_app/utils/video/frame_selection.py,sha256=2x_F6Ez7WHDmvC-FeD6lO31kPvG5oRLrDQIhndoRT-g,4736
60
- lightning_pose_app-1.8.1a18.dist-info/METADATA,sha256=t2lcCQhIsl0259HOR26Zy1SOTSCOUv4-be5qMB59oXw,1464
61
- lightning_pose_app-1.8.1a18.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
62
- lightning_pose_app-1.8.1a18.dist-info/RECORD,,
60
+ lightning_pose_app-1.8.1a19.dist-info/METADATA,sha256=kSKr68pol8TQRgzdfyx_S0zUoMWrH6DHqxZ_Lx71VaY,1464
61
+ lightning_pose_app-1.8.1a19.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
62
+ lightning_pose_app-1.8.1a19.dist-info/RECORD,,
@@ -146,14 +146,19 @@ def _bundle_adjust_impl(
146
146
  f"Insufficient frames found after filtering for session {request.sessionKey}. Possible error in session extraction logic.")
147
147
 
148
148
  # Remove rows with NaN coordinates
149
+ # Get indices of rows with NaN coordinates in any view
150
+ nan_indices = set()
149
151
  for view in views:
150
152
  df = dfs_by_view[view]
151
153
  picked_columns = [c for c in df.columns if c[2] in ("x", "y")]
152
- dfs_by_view[view] = df.loc[:, picked_columns].dropna()
154
+ nan_indices.update(df.loc[:, picked_columns].dropna().index.symmetric_difference(df.index))
155
+
156
+ # Drop those indices from all views
157
+ for view in views:
158
+ dfs_by_view[view] = dfs_by_view[view].drop(nan_indices)
153
159
  if len(dfs_by_view[view]) == 0:
154
160
  raise RuntimeError(
155
161
  f"Insufficient frames found after dropping NaN rows for session {request.sessionKey}.")
156
-
157
162
  # Normalize columns: x, y alternating.
158
163
  for view in views:
159
164
  df = dfs_by_view[view]