oops

SkyTruth · Nov 21, 2024 · 996b82c · 996b82c
1 parent 6e3b44f
commit 996b82c
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/notebooks/ASA_test_environment.py b/notebooks/ASA_test_environment.py
@@ -224,7 +224,7 @@ def plot_coincidence(
     plt.show()
 
 
-analyzers: dict[str, SourceAnalyzer] = {}
+analyzers: dict[int, SourceAnalyzer] = {}
 
 # %%
 slick_ids = [
@@ -243,8 +243,8 @@ def plot_coincidence(
     s1_scene = get_s1_scene(slick_gdf.s1_scene_id.iloc[0])
 
     source_types = []
-    source_types += ["infra"]
-    # source_types += ["ais"]
+    # source_types += [1]  # ais
+    source_types += [2]  # infra
     if not (  # If the last analyzer is for the same scene, reuse it
         analyzers
         and next(iter(analyzers.items()))[1].s1_scene.scene_id == s1_scene.scene_id
@@ -275,8 +275,8 @@ def plot_coincidence(
             ]
         )
 
-    if "infra" in analyzers:
-        plot_coincidence(analyzers["infra"], slick_id)
+    if 2 in analyzers:
+        plot_coincidence(analyzers[2], slick_id)
 
     print(ranked_sources[["type", "st_name", "collated_score"]].head())