Black

rubin_sim/maf/maf_contrib/lss_obs_strategy/galaxy_counts_metric_extended.py

@@ -127,21 +127,23 @@ def _gal_count(self, apparent_mag, coaddm5):
         # calculate the change in the power law constant based on the band
         # colors assumed here: (u-g)=(g-r)=(r-i)=(i-z)= (z-y)=0.4
         factor = 0.4
-        band_correction_dict = {'u': -3.0 * factor,
-                           'g': -2.0 * factor,
-                           'r': -1.0 * factor,
-                           'i': 0.0,
-                           'z': factor,
-                           'y': 2.0 * factor}
+        band_correction_dict = {
+            "u": -3.0 * factor,
+            "g": -2.0 * factor,
+            "r": -1.0 * factor,
+            "i": 0.0,
+            "z": factor,
+            "y": 2.0 * factor,
+        }
         if self.filter_band not in band_correction_dict:
-            warnings.warn("Invalid band in GalaxyCountsMetricExtended. "
-                          "Assuming i-band instead.")
+            warnings.warn("Invalid band in GalaxyCountsMetricExtended. " "Assuming i-band instead.")
         band_correction = band_correction_dict.get(self.filter_band, 0.0)
 
         # check to make sure that the z-bin assigned is valid.
         if (self.redshift_bin != "all") and (self.redshift_bin not in list(self.power_law_const_a.keys())):
-            warnings.warn("Invalid redshift bin in GalaxyCountsMetricExtended. "
-                          "Defaulting to all redshifts.")
+            warnings.warn(
+                "Invalid redshift bin in GalaxyCountsMetricExtended. " "Defaulting to all redshifts."
+            )
             self.redshift_bin = "all"
 
         # consider the power laws

rubin_sim/maf/metric_bundles/metric_bundle.py

@@ -88,7 +88,7 @@ class MetricBundle:
     * constraint (an optional definition of a large subset of data)
 
     Together these define a unique combination of an opsim benchmark,
-    or "metric bundle". 
+    or "metric bundle".
     An example would be:
     a CountMetric, a HealpixSlicer, and a constraint of 'filter="r"'.
 

rubin_sim/maf/metric_bundles/mo_metric_bundle.py

@@ -172,6 +172,7 @@ class MoMetricBundle(MetricBundle):
     opsim run name, and relevant stackers and maps
     to apply when calculating the metric values.
     """
+
     def __init__(
         self,
         metric,
@@ -373,6 +374,7 @@ class MoMetricBundleGroup:
     verbose : `bool`, opt
         Flag to turn on/off verbose feedback.
     """
+
     def __init__(self, bundle_dict, out_dir=".", results_db=None, verbose=True):
         self.verbose = verbose
         self.bundle_dict = bundle_dict

rubin_sim/maf/metrics/mo_metrics.py

@@ -266,6 +266,7 @@ class NObsNoSinglesMetric(BaseMoMetric):
         determined using the _calcVis method (completeness calculation).
         If not None, the snr is calculated and used as a flat cutoff instead.
     """
+
     def __init__(self, snr_limit=None, **kwargs):
         super().__init__(**kwargs)
         self.snr_limit = snr_limit
@@ -291,6 +292,7 @@ class NNightsMetric(BaseMoMetric):
         determined using the _calcVis method (completeness calculation).
         If not None, the snr is calculated and used as a flat cutoff instead.
     """
+
     def __init__(self, snr_limit=None, **kwargs):
         super().__init__(**kwargs)
         self.snr_limit = snr_limit
@@ -314,6 +316,7 @@ class ObsArcMetric(BaseMoMetric):
         determined using the _calcVis method (completeness calculation).
         If not None, the snr is calculated and used as a flat cutoff instead.
     """
+
     def __init__(self, snr_limit=None, **kwargs):
         super().__init__(**kwargs)
         self.snr_limit = snr_limit
@@ -487,6 +490,7 @@ class DiscoveryNChancesMetric(BaseChildMetric):
     badval : `float`, opt
         Value to return when metric cannot be calculated.
     """
+
     def __init__(
         self,
         parent_discovery_metric,
@@ -524,6 +528,7 @@ class DiscoveryNObsMetric(BaseChildMetric):
     badval : `float`, opt
         Value to return when metric cannot be calculated.
     """
+
     def __init__(self, parent_discovery_metric, badval=0, **kwargs):
         super().__init__(parent_discovery_metric, badval=badval, **kwargs)
         # The number of the discovery chance to use.