From 929da17c1917d28f86bd53a250d2b69a4cfe5544 Mon Sep 17 00:00:00 2001 From: pranmod01 Date: Fri, 18 Oct 2024 09:10:00 -0700 Subject: [PATCH] formatting fixed --- docs/how_to_guide/plot_02_glm_demo.py | 2 +- docs/how_to_guide/plot_03_glm_pytree.py | 4 ++-- docs/how_to_guide/plot_04_population_glm.py | 2 +- docs/tutorials/plot_01_current_injection.py | 4 ++-- docs/tutorials/plot_06_calcium_imaging.py | 4 ++-- src/nemos/__init__.py | 2 +- src/nemos/_documentation_utils/__init__.py | 2 +- 7 files changed, 10 insertions(+), 10 deletions(-) diff --git a/docs/how_to_guide/plot_02_glm_demo.py b/docs/how_to_guide/plot_02_glm_demo.py index 23ef1396..f1c6e3b2 100644 --- a/docs/how_to_guide/plot_02_glm_demo.py +++ b/docs/how_to_guide/plot_02_glm_demo.py @@ -25,12 +25,12 @@ """ +import jax import matplotlib.pyplot as plt import numpy as np from matplotlib.patches import Rectangle from sklearn import model_selection -import jax import nemos as nmo np.random.seed(111) diff --git a/docs/how_to_guide/plot_03_glm_pytree.py b/docs/how_to_guide/plot_03_glm_pytree.py index 22e0f5f2..2d36db3b 100644 --- a/docs/how_to_guide/plot_03_glm_pytree.py +++ b/docs/how_to_guide/plot_03_glm_pytree.py @@ -12,10 +12,10 @@ First, however, let's briefly discuss FeaturePytrees. """ -import numpy as np - import jax import jax.numpy as jnp +import numpy as np + import nemos as nmo np.random.seed(111) diff --git a/docs/how_to_guide/plot_04_population_glm.py b/docs/how_to_guide/plot_04_population_glm.py index a4e0eec4..70dac9cd 100644 --- a/docs/how_to_guide/plot_04_population_glm.py +++ b/docs/how_to_guide/plot_04_population_glm.py @@ -22,10 +22,10 @@ Let's generate some synthetic data and fit a population model. """ +import jax.numpy as jnp import matplotlib.pyplot as plt import numpy as np -import jax.numpy as jnp import nemos as nmo np.random.seed(123) diff --git a/docs/tutorials/plot_01_current_injection.py b/docs/tutorials/plot_01_current_injection.py index aeafe377..8eb83dbe 100644 --- a/docs/tutorials/plot_01_current_injection.py +++ b/docs/tutorials/plot_01_current_injection.py @@ -48,12 +48,12 @@ +# Import everything +import jax import matplotlib.pyplot as plt import numpy as np import pynapple as nap -# Import everything -import jax import nemos as nmo # some helper plotting functions diff --git a/docs/tutorials/plot_06_calcium_imaging.py b/docs/tutorials/plot_06_calcium_imaging.py index 514a0367..985a0b61 100644 --- a/docs/tutorials/plot_06_calcium_imaging.py +++ b/docs/tutorials/plot_06_calcium_imaging.py @@ -10,12 +10,12 @@ """ +import jax +import jax.numpy as jnp import matplotlib.pyplot as plt import pynapple as nap from sklearn.linear_model import LinearRegression -import jax -import jax.numpy as jnp import nemos as nmo # %% diff --git a/src/nemos/__init__.py b/src/nemos/__init__.py index aedd05c0..97c5b3db 100644 --- a/src/nemos/__init__.py +++ b/src/nemos/__init__.py @@ -14,5 +14,5 @@ styles, tree_utils, type_casting, - utils + utils, ) diff --git a/src/nemos/_documentation_utils/__init__.py b/src/nemos/_documentation_utils/__init__.py index 1c64a43a..3cd63e0e 100644 --- a/src/nemos/_documentation_utils/__init__.py +++ b/src/nemos/_documentation_utils/__init__.py @@ -19,5 +19,5 @@ plot_rates_and_smoothed_counts, plot_weighted_sum_basis, run_animation, - tuning_curve_plot + tuning_curve_plot, )