format codes

hsulab · Apr 19, 2024 · 4403446 · 4403446
1 parent 0ada113
commit 4403446
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/src/gdpx/potential/__init__.py b/src/gdpx/potential/__init__.py
@@ -9,7 +9,8 @@
 
 from .. import config
 from ..core.register import registers
+from ..bias import bias_register
 
 
 if __name__ == "__main__":
-    ...
+    ...
diff --git a/src/gdpx/potential/managers/bias.py b/src/gdpx/potential/managers/bias.py
@@ -6,9 +6,9 @@
 
 from . import registers
 from . import AbstractPotentialManager, DummyCalculator
-from gdpx.bias import bias_register
+from .. import bias_register
 
-"""This manager registers ALL jax-based bias calculators."""
+"""This manager registers ALL bias calculators."""
 
 
 class BiasManager(AbstractPotentialManager):
@@ -26,7 +26,7 @@ def __init__(self) -> None:
         super().__init__()
 
         return
-    
+
     def register_calculator(self, calc_params, *agrs, **kwargs) -> None:
         """"""
         super().register_calculator(calc_params, *agrs, **kwargs)
@@ -54,11 +54,11 @@ def register_calculator(self, calc_params, *agrs, **kwargs) -> None:
             calc = bias_register[bias_type](**calc_params)
         else:
             ...
-        
+
         self.calc = calc
 
         return
 
 
 if __name__ == "__main__":
-    ...
+    ...