chore: black formatting

alercebroker · Jan 17, 2024 · c672564 · c672564
1 parent 25831e3
commit c672564
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/magstats_step/magstats_step/core/_base.py b/magstats_step/magstats_step/core/_base.py
@@ -21,7 +21,7 @@ def __init__(self, detections: List[dict]):
         except KeyError:  # extra_fields is not present
             self._detections = pd.DataFrame.from_records(detections)
         self._detections = self._detections.drop_duplicates(
-           ["candid", "oid"]
+            ["candid", "oid"]
         ).set_index("candid")
         # Select only non-forced detections
         self._detections = self._detections[~self._detections["forced"]]

diff --git a/magstats_step/magstats_step/step.py b/magstats_step/magstats_step/step.py
@@ -38,7 +38,9 @@ def _execute(self, messages: dict):
         magstats = magstats.set_index("oid").replace({np.nan: None})
         for oid in stats:
             try:
-                stats[oid]["magstats"] = magstats.loc[oid].iloc[0].to_dict("records")
+                stats[oid]["magstats"] = (
+                    magstats.loc[oid].iloc[0].to_dict("records")
+                )
             except TypeError:
                 stats[oid]["magstats"] = [magstats.loc[oid].iloc[0].to_dict()]
 
@@ -66,7 +68,9 @@ def _execute_ztf(self, messages: dict):
         magstats = magstats.set_index("oid").replace({np.nan: None})
         for oid in stats:
             try:
-                stats[oid]["magstats"] = magstats.loc[oid].iloc[0].to_dict("records")
+                stats[oid]["magstats"] = (
+                    magstats.loc[oid].iloc[0].to_dict("records")
+                )
                 # el primero o uno por par oid_candid?
             except TypeError:
                 stats[oid]["magstats"] = [magstats.loc[oid].iloc[0].to_dict()]