pre-commit fixes

MiraGeoscience · Feb 12, 2024 · 0ee8145 · 0ee8145
1 parent 2f262ad
commit 0ee8145
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 7 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -33,6 +33,7 @@ repos:
     hooks:
     -   id: pyupgrade
         args: [--py39-plus]
+        exclude: plates
 -   repo: https://github.com/pre-commit/mirrors-mypy
     rev: v1.6.1
     hooks:

diff --git a/plate_simulation/models/plates.py b/plate_simulation/models/plates.py
@@ -8,6 +8,7 @@
 from __future__ import annotations
 
 from collections.abc import Sequence
+from typing import Optional
 
 import numpy as np
 from geoapps_utils.transformations import rotate_xyz
@@ -45,7 +46,7 @@ class Plate(BaseModel):
     dip: float = 0.0
     azimuth: float = 0.0
     reference: str = "center"
-    _surface: Surface | None = None
+    _surface: Optional[Surface] = None
 
     @classmethod
     def from_params(cls, params: PlateParams):

diff --git a/tests/models/plates_test.py b/tests/models/plates_test.py
@@ -40,14 +40,17 @@ def test_plate(tmp_path):
         500.0,
     )
     assert (
-        vertical_striking_north.vertices[:, 0].mean() == plate.center_x
-    )  # pylint: disable=no-member
+        vertical_striking_north.vertices[:, 0].mean()
+        == plate.center_x  # pylint: disable=no-member
+    )
     assert (
-        vertical_striking_north.vertices[:, 1].mean() == plate.center_y
-    )  # pylint: disable=no-member
+        vertical_striking_north.vertices[:, 1].mean()
+        == plate.center_y  # pylint: disable=no-member
+    )
     assert (
-        vertical_striking_north.vertices[:, 2].mean() == plate.center_z
-    )  # pylint: disable=no-member
+        vertical_striking_north.vertices[:, 2].mean()
+        == plate.center_z  # pylint: disable=no-member
+    )
 
     plate = Plate(
         name="my other plate",