diff --git a/neuralpredictors/layers/readouts/gaussian.py b/neuralpredictors/layers/readouts/gaussian.py index 3023b5e6..656183b1 100644 --- a/neuralpredictors/layers/readouts/gaussian.py +++ b/neuralpredictors/layers/readouts/gaussian.py @@ -294,9 +294,7 @@ def __init__( torch.normal(mean=torch.ones(1, outdims), std=torch.ones(1, outdims)) ) elif self._regularizer_type != "l1": - raise ValueError( - f"regularizer_type should be 'l1' or 'adaptive_log_norm' but got {self._regularizer_type}" - ) + raise ValueError(f"regularizer_type should be 'l1' or 'adaptive_log_norm' but got {self._regularizer_type}") if init_mu_range > 1.0 or init_mu_range <= 0.0 or init_sigma <= 0.0: raise ValueError("either init_mu_range doesn't belong to [0.0, 1.0] or init_sigma_range is non-positive")