refactor(verification): reorganize imports and function structure

notebooks/verification.py

@@ -13,20 +13,24 @@ app = marimo.App(width="medium", app_title="Pipeline Verification")
 
 
 @app.cell
-def import_packages():
+def _():
-    from io import BytesIO
-
     import marimo as mo
+
+    return (mo,)
+
+
+@app.cell
+def global_base_deps():
     import numpy as np
     import polars as pl
-    from habitat.utils.visualizations import maps
-    from matplotlib import pyplot as plt
     from PIL import Image
 
-    from rich.progress import track
+    return (Image, np, pl)
 
+
+@app.cell
+def project_setup():
     from configs import cfg_manager
-    from compressors.pipeline import HashPipeline
     from scenegraph import ObjectNode, RoomNode, SimpleSceneGraph
     from simulator import (
         HabitatSimulatorConfig,
@@ -34,35 +38,18 @@ def import_packages():
         collect_room_views_by_room,
         create_habitat_simulator,
         render_topdown_scene_map,
-        save_object_image,
-        save_room_view,
     )
-    from compressors.proposal import extract_masked_region, generate_proposals_batch
-    from utils import numpy_to_pil
 
     return (
         HabitatSimulatorConfig,
-        HashPipeline,
-        Image,
         ObjectNode,
         RoomNode,
         SimpleSceneGraph,
         TopDownSceneElements,
+        cfg_manager,
         collect_room_views_by_room,
         create_habitat_simulator,
-        cfg_manager,
-        extract_masked_region,
-        maps,
-        mo,
-        np,
-        numpy_to_pil,
-        pl,
-        plt,
         render_topdown_scene_map,
-        save_object_image,
-        save_room_view,
-        generate_proposals_batch,
-        track,
     )
 
 
@@ -125,13 +112,14 @@ def setup_verification_context(
 @app.cell
 def render_topdown_room_map(
     TopDownSceneElements,
-    maps,
     meters_per_pixel,
-    plt,
     render_topdown_scene_map,
     room_nodes,
     sim,
 ):
+    from habitat.utils.visualizations import maps
+    from matplotlib import pyplot as plt
+
     render_topdown_scene_map(
         pathfinder=sim.pathfinder,
         elements=TopDownSceneElements(room_nodes=room_nodes),
@@ -144,29 +132,28 @@ def render_topdown_room_map(
 
 @app.cell
 def build_scene_graph_pipeline(
-    HashPipeline,
     Image,
     ObjectNode,
     SimpleSceneGraph,
     agent,
-    collect_room_views_by_room,
     cfg_manager,
-    extract_masked_region,
+    collect_room_views_by_room,
     hash_bits,
-    mo,
     np,
-    numpy_to_pil,
     pipeline_batch_size,
     room_nodes,
     sam_max_masks,
     sam_min_area,
-    generate_proposals_batch,
-    save_object_image,
-    save_room_view,
     sim,
-    track,
     views_per_room,
 ):
+    from rich.progress import track
+
+    from compressors.pipeline import HashPipeline
+    from compressors.proposal import extract_masked_region, generate_proposals_batch
+    from simulator import save_object_image, save_room_view
+    from utils import numpy_to_pil
+
     all_room_views = collect_room_views_by_room(
         agent=agent,
         sim=sim,