diff --git a/src/pspec_likelihood/__init__.py b/src/pspec_likelihood/__init__.py index 2b9ec11..0fe595a 100644 --- a/src/pspec_likelihood/__init__.py +++ b/src/pspec_likelihood/__init__.py @@ -1,4 +1,5 @@ """The PSpec Likelihood Package.""" + from . import likelihood from .arbitrary_linear_systematics import LikelihoodLinearSystematic from .likelihood import ( diff --git a/src/pspec_likelihood/arbitrary_linear_systematics.py b/src/pspec_likelihood/arbitrary_linear_systematics.py index eb121ba..7f91b0e 100644 --- a/src/pspec_likelihood/arbitrary_linear_systematics.py +++ b/src/pspec_likelihood/arbitrary_linear_systematics.py @@ -1,4 +1,5 @@ """Module responsible for computing the likelihood for linear systematics.""" + from __future__ import annotations import warnings diff --git a/src/pspec_likelihood/likelihood.py b/src/pspec_likelihood/likelihood.py index c95766b..cf29852 100644 --- a/src/pspec_likelihood/likelihood.py +++ b/src/pspec_likelihood/likelihood.py @@ -1,4 +1,5 @@ """Primary module defining likelihoods based on HERA power spectra.""" + from __future__ import annotations import warnings @@ -733,12 +734,16 @@ def loglike(self, theory_params, sys_params) -> float: # them. "basis" corresponds to "A" in the memo. basis = self.linear_systematics_basis_function( sys_params, - self.model.kperp_bins_theory - if self.model.apply_window_to_systematics - else self.model.kperp_bins_obs, - self.model.kpar_bins_theory - if self.model.apply_window_to_systematics - else self.model.kpar_bins_obs, + ( + self.model.kperp_bins_theory + if self.model.apply_window_to_systematics + else self.model.kperp_bins_obs + ), + ( + self.model.kpar_bins_theory + if self.model.apply_window_to_systematics + else self.model.kpar_bins_obs + ), ) mu_linear = self.get_mu_linear(basis) diff --git a/src/pspec_likelihood/types.py b/src/pspec_likelihood/types.py index 852126d..7225339 100644 --- a/src/pspec_likelihood/types.py +++ b/src/pspec_likelihood/types.py @@ -1,4 +1,5 @@ """Types to be used throughout the package.""" + from __future__ import annotations from typing import Any, Callable diff --git a/src/pspec_likelihood/utils.py b/src/pspec_likelihood/utils.py index d21e318..016b27e 100644 --- a/src/pspec_likelihood/utils.py +++ b/src/pspec_likelihood/utils.py @@ -1,4 +1,5 @@ """Miscellaneous utility functions.""" + from __future__ import annotations import warnings diff --git a/tests/test_likelihoods.py b/tests/test_likelihoods.py index abba301..ab40e4b 100644 --- a/tests/test_likelihoods.py +++ b/tests/test_likelihoods.py @@ -1,4 +1,5 @@ """Test loading an UVPspec file""" + from __future__ import annotations import astropy.units as un diff --git a/tests/test_uvpread.py b/tests/test_uvpread.py index c577fe7..315704a 100644 --- a/tests/test_uvpread.py +++ b/tests/test_uvpread.py @@ -1,4 +1,5 @@ """Test loading an UVPspec file""" + from pathlib import Path import astropy.units as un