From ce34d01e19581dcbc35893d1ea0f445246481268 Mon Sep 17 00:00:00 2001 From: Mehdy Lefkir Date: Mon, 2 Sep 2024 18:14:37 +0100 Subject: [PATCH] last pep8 issues fixed? --- stingray/modeling/gpmodeling.py | 2 +- stingray/modeling/tests/test_gpmodeling.py | 29 +++++++++++----------- 2 files changed, 16 insertions(+), 15 deletions(-) diff --git a/stingray/modeling/gpmodeling.py b/stingray/modeling/gpmodeling.py index f57be4836..39a5f5305 100644 --- a/stingray/modeling/gpmodeling.py +++ b/stingray/modeling/gpmodeling.py @@ -288,7 +288,7 @@ def run_prior_checks( approximate_with="SHO", ): """Check the approximation of the power spectrum density. - This function will plot various diagnostics on the residuals and the ratio + This function will plot various diagnostics on the residuals and the ratio of the PSD and the approximate PSD. Parameters diff --git a/stingray/modeling/tests/test_gpmodeling.py b/stingray/modeling/tests/test_gpmodeling.py index 84beba24d..76f34fc76 100644 --- a/stingray/modeling/tests/test_gpmodeling.py +++ b/stingray/modeling/tests/test_gpmodeling.py @@ -3,6 +3,21 @@ import numpy as np import matplotlib.pyplot as plt +from stingray.modeling.gpmodeling import ( + get_kernel, + get_mean, + get_gp_params, + _psd_model, + get_psd_and_approx, + run_prior_checks, + _get_coefficients_approximation, + get_prior, + get_log_likelihood, + GPResult, + get_priors_samples, +) +from stingray import Lightcurve + try: import jax import jax.numpy as jnp @@ -22,20 +37,6 @@ except ImportError: _HAS_TINYGP = False -from stingray.modeling.gpmodeling import ( - get_kernel, - get_mean, - get_gp_params, - _psd_model, - get_psd_and_approx, - run_prior_checks, - _get_coefficients_approximation, - get_prior, - get_log_likelihood, - GPResult, - get_priors_samples, -) -from stingray import Lightcurve try: import tensorflow_probability.substrates.jax as tfp