Ruff

Author: jeanconn

proseco/catalog.py

@@ -447,6 +447,7 @@ def optimize_acqs_fids(self, initial_guide_cands=None, **kwargs):
 
         # Only do these imports if it turns out we need optimization.
         from chandra_aca.star_probs import guide_count
+
         from . import characteristics_acq as ACQ
 
         no_fid_guides = get_guide_catalog(

proseco/fid.py

@@ -485,7 +485,6 @@ def set_spoilers_score(self, fid):
 
         # Run guide star fid_trap checks if guide candidates are available
         if self.guide_cands is not None:
-
             fid_trap, _ = guide.check_fid_trap(self.guide_cands, [fid], dither)
             if np.any(fid_trap):
                 fid["fid_trap_spoiler"] = True

proseco/guide.py

@@ -151,7 +151,7 @@ def get_guide_catalog(obsid=0, initial_guide_cands=None, **kwargs):
         # for any previous run
         guides.cand_guides["stage"] = -1
         for stage_idx in range(len(GUIDE.stages)):
-            guides.cand_guides[f"stat_{stage_idx+1}"] = 0
+            guides.cand_guides[f"stat_{stage_idx + 1}"] = 0
 
         # Refilter candidates for the current fid set if needed
         if guides.fids is not None and len(guides.fids) > 0:
@@ -776,7 +776,7 @@ def search_stage(self, stage):
 
         # Adopt the SAUSAGE convention of a bit array for errors
         # Not all items will be checked for each star (allow short circuit)
-        scol =  f"stat_{stage["Stage"]}"
+        scol = f"stat_{stage['Stage']}"
         cand_guides[scol] = 0
 
         n_sigma = stage["SigErrMultiplier"]
@@ -936,7 +936,7 @@ def process_include_ids(self, cand_guides, stars):
         # Now compute imposter mags for these stars
         if len(new_stars) > 0:
             imp_mag, imp_row, imp_col = get_imposter_mags(
-            new_stars, self.dark, self.dither
+                new_stars, self.dark, self.dither
             )
             new_stars["imp_mag"] = imp_mag
             new_stars["imp_r"] = imp_row

proseco/tests/test_mon_full_cat.py

@@ -196,27 +196,31 @@ def test_mon_takes_guide():
 
 
 def test_obsid_30843():
-    args = {'obsid': 30843.0,
-        'raise_exc': True,
-        'att': [-0.50655663, -0.04177606, 0.85826295, 0.07099192],
-        'date': '2025:074:22:39:37.088',
-        'detector': 'ACIS-S',
-        'dither_acq': [15.9984, 15.9984],
-        'dither_guide': [15.9984, 15.9984],
-        'focus_offset': 0.0,
-        'man_angle': 124.90441066524069,
-        'n_acq': 8, 'n_fid': 3, 'n_guide': 5,
-        'sim_offset': -2442.0,
-        't_ccd_acq': -6.367734460876512,
-        't_ccd_guide': -3.9378722816242115,
-        'dyn_bgd_n_faint': 2,
-        'dyn_bgd_dt_ccd': -4.0,
-        'target_offset': [0., 0.],
-        'target_name': 'WR 25',
-        'duration': 39000.0,
-        'man_angle_next': 157.2577020420281,
-        'monitors': [[161.043292, -59.719753, 0., 8.8, 0.]]}
+    args = {
+        "obsid": 30843.0,
+        "raise_exc": True,
+        "att": [-0.50655663, -0.04177606, 0.85826295, 0.07099192],
+        "date": "2025:074:22:39:37.088",
+        "detector": "ACIS-S",
+        "dither_acq": [15.9984, 15.9984],
+        "dither_guide": [15.9984, 15.9984],
+        "focus_offset": 0.0,
+        "man_angle": 124.90441066524069,
+        "n_acq": 8,
+        "n_fid": 3,
+        "n_guide": 5,
+        "sim_offset": -2442.0,
+        "t_ccd_acq": -6.367734460876512,
+        "t_ccd_guide": -3.9378722816242115,
+        "dyn_bgd_n_faint": 2,
+        "dyn_bgd_dt_ccd": -4.0,
+        "target_offset": [0.0, 0.0],
+        "target_name": "WR 25",
+        "duration": 39000.0,
+        "man_angle_next": 157.2577020420281,
+        "monitors": [[161.043292, -59.719753, 0.0, 8.8, 0.0]],
+    }
     aca = get_aca_catalog(**args)
     mon_id = 1130642984
-    assert mon_id in aca['id']
-    assert mon_id in aca.mons['id']
+    assert mon_id in aca["id"]
+    assert mon_id in aca.mons["id"]