Fix ruff formatting

Author: C1587S

src/dscim/menu/main_recipe.py

@@ -247,17 +247,17 @@ def __init__(
         self.logger = logging.getLogger(__name__)
 
         if self.quantreg_quantiles is not None:
-            assert len(self.quantreg_quantiles) == len(
-                self.quantreg_weights
-            ), "Length of quantreg quantiles does not match length of weights."
+            assert len(self.quantreg_quantiles) == len(self.quantreg_weights), (
+                "Length of quantreg quantiles does not match length of weights."
+            )
 
-        assert (
-            self.discounting_type in self.DISCOUNT_TYPES
-        ), f"Discount type not implemented. Try one of {self.DISCOUNT_TYPES}."
+        assert self.discounting_type in self.DISCOUNT_TYPES, (
+            f"Discount type not implemented. Try one of {self.DISCOUNT_TYPES}."
+        )
 
-        assert (
-            self.formula in self.FORMULAS
-        ), f"Formula not implemented. Try one of {self.FORMULAS}."
+        assert self.formula in self.FORMULAS, (
+            f"Formula not implemented. Try one of {self.FORMULAS}."
+        )
 
         # Set stream of discounts to None if discounting_type is 'constant'
         # 'constant_model_collapsed' should be here except that we allow
@@ -488,10 +488,7 @@ def _aggregate_by_geography(
             return data  # No aggregation needed
 
         elif geography == "country":
-            if (
-                hasattr(self, "country_mapping")
-                and self.country_mapping is not None
-            ):
+            if hasattr(self, "country_mapping") and self.country_mapping is not None:
                 return self._aggregate_to_country(data)
             else:
                 raise ValueError(
@@ -530,9 +527,7 @@ def _aggregate_to_country(self, data: xr.DataArray) -> xr.DataArray:
             else "nopop"
             for _, row in self.country_mapping.iterrows()
         }
-        territories = [
-            mapping.get(str(r)[:3], "unknown") for r in data.region.values
-        ]
+        territories = [mapping.get(str(r)[:3], "unknown") for r in data.region.values]
         return data.assign_coords({"region": territories}).groupby("region").sum()
 
     def damages_dataset(self, geography: str = "globe") -> xr.Dataset:
@@ -579,9 +574,7 @@ def _filter_illegal_combinations_xr(self, ds: xr.Dataset) -> xr.Dataset:
         xr.Dataset
             Filtered dataset with illegal combinations set to NaN
         """
-        if "ssp" in ds.coords and any(
-            ssp in ds.ssp.values for ssp in ["SSP1", "SSP5"]
-        ):
+        if "ssp" in ds.coords and any(ssp in ds.ssp.values for ssp in ["SSP1", "SSP5"]):
             self.logger.info("Dropping illegal model combinations.")
             illegal = ((ds.ssp == "SSP1") & (ds.rcp == "rcp85")) | (
                 (ds.ssp == "SSP5") & (ds.rcp == "rcp45")

tests/test_geography.py

@@ -139,7 +139,11 @@ def test_xarray_path_matches_pandas_path_for_globe(self, menu_instance):
         assert "damages" in pandas_result.columns
         assert "damages" in xarray_result.columns
 
-        sort_cols = [c for c in ["year", "ssp", "model", "gcm", "rcp"] if c in pandas_result.columns]
+        sort_cols = [
+            c
+            for c in ["year", "ssp", "model", "gcm", "rcp"]
+            if c in pandas_result.columns
+        ]
         pandas_sorted = pandas_result.sort_values(sort_cols).reset_index(drop=True)
         xarray_sorted = xarray_result.sort_values(sort_cols).reset_index(drop=True)