Fixed bug with grid square registration not running if the CLEM atlas is the last dataset to be registered

Author: tieneupin

src/murfey/workflows/clem/register_preprocessing_results.py

@@ -501,13 +501,12 @@ def run(message: dict, murfey_db: Session, demo: bool = False) -> dict[str, bool
             )
 
         try:
-            # Register dataset as grid square
-            if "Overview_" not in result.series_name:
-                _register_grid_square(
-                    session_id=session_id,
-                    result=result,
-                    murfey_db=murfey_db,
-                )
+            # Register CLEM image series as grid squares
+            _register_grid_square(
+                session_id=session_id,
+                result=result,
+                murfey_db=murfey_db,
+            )
         except Exception:
             # Log error but allow workflow to proceed
             logger.error(

tests/workflows/clem/test_register_preprocessing_results.py

@@ -160,7 +160,7 @@ def test_run(
         assert result == {"success": True}
     assert mock_register_clem_series.call_count == len(preprocessing_messages)
     assert mock_register_dcg_and_atlas.call_count == len(preprocessing_messages)
-    assert mock_register_grid_square.call_count == len(preprocessing_messages) - 1
+    assert mock_register_grid_square.call_count == len(preprocessing_messages)
     assert mock_align_and_merge_call.call_count == len(preprocessing_messages) * len(
         colors
     )