lightning-pose-app 1.8.1a19__py3-none-any.whl → 1.8.1a20__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.1a19
3
+ Version: 1.8.1a20
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=dRLPh9zb4tvqVU87mP4SWPx00D7dmzx34s2c0hznN3g,8296
35
+ litpose_app/routes/labeler/bundle_adjust.py,sha256=c5phH1ufPAKOnMwrv8cfqeSITdaetdPNDlKMaqsnaR0,8378
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.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,,
60
+ lightning_pose_app-1.8.1a20.dist-info/METADATA,sha256=yad6hrvpLGwr6YtrEgdgVY4z0_q3bKqN-mGcMLklem8,1464
61
+ lightning_pose_app-1.8.1a20.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
62
+ lightning_pose_app-1.8.1a20.dist-info/RECORD,,
@@ -144,18 +144,18 @@ def _bundle_adjust_impl(
144
144
  if len(dfs_by_view[view]) == 0:
145
145
  raise RuntimeError(
146
146
  f"Insufficient frames found after filtering for session {request.sessionKey}. Possible error in session extraction logic.")
147
-
148
147
  # Remove rows with NaN coordinates
149
148
  # Get indices of rows with NaN coordinates in any view
150
149
  nan_indices = set()
151
150
  for view in views:
152
151
  df = dfs_by_view[view]
153
152
  picked_columns = [c for c in df.columns if c[2] in ("x", "y")]
154
- nan_indices.update(df.loc[:, picked_columns].dropna().index.symmetric_difference(df.index))
153
+ nan_rows = df.loc[:, picked_columns].dropna().index.symmetric_difference(df.index)
154
+ nan_indices.update(df.index.get_indexer(nan_rows))
155
155
 
156
- # Drop those indices from all views
156
+ # Drop those indices from all views
157
157
  for view in views:
158
- dfs_by_view[view] = dfs_by_view[view].drop(nan_indices)
158
+ dfs_by_view[view] = dfs_by_view[view].drop(dfs_by_view[view].index[list(nan_indices)])
159
159
  if len(dfs_by_view[view]) == 0:
160
160
  raise RuntimeError(
161
161
  f"Insufficient frames found after dropping NaN rows for session {request.sessionKey}.")