diff --git a/neurometry/curvature/datasets/gridcells.py b/neurometry/curvature/datasets/gridcells.py
index a152346..b2a9afa 100644
--- a/neurometry/curvature/datasets/gridcells.py
+++ b/neurometry/curvature/datasets/gridcells.py
@@ -6,7 +6,7 @@
 import neurometry.curvature.datasets.structures as structures
 
 os.environ["GEOMSTATS_BACKEND"] = "pytorch"
-import geomstats.backend as gs  # noqa: E402
+import geomstats.backend as gs
 
 
 # TODO
diff --git a/neurometry/curvature/datasets/synthetic.py b/neurometry/curvature/datasets/synthetic.py
index 45f6bef..89f73a2 100644
--- a/neurometry/curvature/datasets/synthetic.py
+++ b/neurometry/curvature/datasets/synthetic.py
@@ -15,8 +15,8 @@
 )
 
 os.environ["GEOMSTATS_BACKEND"] = "pytorch"
-import geomstats.backend as gs  # noqa: E402
-from geomstats.geometry.special_orthogonal import SpecialOrthogonal  # noqa: E402
+import geomstats.backend as gs
+from geomstats.geometry.special_orthogonal import SpecialOrthogonal
 
 
 def load_projected_images(n_scalars=5, n_angles=1000, img_size=128):
diff --git a/neurometry/curvature/evaluate.py b/neurometry/curvature/evaluate.py
index 65f4b8a..35eb407 100644
--- a/neurometry/curvature/evaluate.py
+++ b/neurometry/curvature/evaluate.py
@@ -11,11 +11,11 @@
 )
 
 os.environ["GEOMSTATS_BACKEND"] = "pytorch"
-import geomstats.backend as gs  # noqa: E402
-from geomstats.geometry.base import ImmersedSet  # noqa: E402
-from geomstats.geometry.euclidean import Euclidean  # noqa: E402
-from geomstats.geometry.pullback_metric import PullbackMetric  # noqa: E402
-from geomstats.geometry.special_orthogonal import SpecialOrthogonal  # noqa: E402
+import geomstats.backend as gs
+from geomstats.geometry.base import ImmersedSet
+from geomstats.geometry.euclidean import Euclidean
+from geomstats.geometry.pullback_metric import PullbackMetric
+from geomstats.geometry.special_orthogonal import SpecialOrthogonal
 
 
 class NeuralManifoldIntrinsic(ImmersedSet):
diff --git a/neurometry/datasets/synthetic.py b/neurometry/datasets/synthetic.py
index 404cbb8..1aa4353 100644
--- a/neurometry/datasets/synthetic.py
+++ b/neurometry/datasets/synthetic.py
@@ -3,11 +3,11 @@
 import torch
 
 os.environ["GEOMSTATS_BACKEND"] = "pytorch"
-import geomstats.backend as gs  # noqa: E402
-from geomstats.geometry.euclidean import Euclidean  # noqa: E402
-from geomstats.geometry.hypersphere import Hypersphere  # noqa: E402
-from geomstats.geometry.klein_bottle import KleinBottle  # noqa: E402
-from geomstats.geometry.product_manifold import ProductManifold  # noqa: E402
+import geomstats.backend as gs
+from geomstats.geometry.euclidean import Euclidean
+from geomstats.geometry.hypersphere import Hypersphere
+from geomstats.geometry.klein_bottle import KleinBottle
+from geomstats.geometry.product_manifold import ProductManifold
 
 
 def synthetic_neural_manifold(
diff --git a/neurometry/dimension/dimension.py b/neurometry/dimension/dimension.py
index 0e65d1d..4176a87 100644
--- a/neurometry/dimension/dimension.py
+++ b/neurometry/dimension/dimension.py
@@ -13,7 +13,7 @@
 import neurometry.datasets.synthetic as synthetic
 
 os.environ["GEOMSTATS_BACKEND"] = "pytorch"
-import geomstats.backend as gs  # noqa: E402
+import geomstats.backend as gs
 
 
 def skdim_dimension_estimation(
diff --git a/pyproject.toml b/pyproject.toml
index bc3f971..3e4448f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -112,6 +112,7 @@ select = [
 ]
 ignore = [
     "E501",    # line too long
+    "E402",    # ignore module level import not at top of file
     "PERF203", # allow try-except within loops
     "RUF012",  # force typing
 ]