formatting fixed

flatironinstitute · Oct 18, 2024 · 929da17 · 929da17
1 parent a9799a3
commit 929da17
Show file tree

Hide file tree

Showing 7 changed files with 10 additions and 10 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -19,5 +19,5 @@
     plot_rates_and_smoothed_counts,
     plot_weighted_sum_basis,
     run_animation,
-    tuning_curve_plot
+    tuning_curve_plot,
 )
-Original file line number
+Diff line change
@@ Expand Up / @@ -14,5 +14,5 @@ @@
         styles,
         tree_utils,
         type_casting,
-        utils
+        utils,
     )