move to fix pre-commit

stfc · Nov 1, 2024 · 9c6e141 · 9c6e141
1 parent 733eb6b
commit 9c6e141
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/janus_core/helpers/observables.py b/janus_core/helpers/observables.py
@@ -147,7 +147,6 @@ def _set_components(self, components: list[str]):
 
 # pylint: disable=too-few-public-methods
 class Stress(Observable, ComponentMixin):
-    __module__ = "observables"
     """
     Observable for stress components.
 
@@ -161,6 +160,8 @@ class Stress(Observable, ComponentMixin):
         Calculate with the ideal gas contribution.
     """
 
+    __module__ = "observables"
+
     def __init__(
         self,
         *,
@@ -234,7 +235,6 @@ def __call__(self, atoms: Atoms) -> list[float]:
 
 # pylint: disable=too-few-public-methods
 class Velocity(Observable, ComponentMixin):
-    __module__ = "observables"
     """
     Observable for per atom velocity components.
 
@@ -246,6 +246,8 @@ class Velocity(Observable, ComponentMixin):
         List or slice of atoms to observe velocities from.
     """
 
+    __module__ = "observables"
+
     def __init__(
         self,
         *,