[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
mosdef-hub · Dec 20, 2023 · 98f04c3 · 98f04c3
1 parent e86a6c5
commit 98f04c3
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 10 deletions.
diff --git a/gmso/core/atom_type.py b/gmso/core/atom_type.py
@@ -7,14 +7,14 @@
 
 from gmso.abc.serialization_utils import unyt_to_dict
 from gmso.core.parametric_potential import ParametricPotential
-from gmso.utils.units import GMSO_UnitRegistry
 from gmso.utils._constants import UNIT_WARNING_STRING
 from gmso.utils.expression import PotentialExpression
 from gmso.utils.misc import (
     ensure_valid_dimensions,
     unyt_compare,
     unyt_to_hashable,
 )
+from gmso.utils.units import GMSO_UnitRegistry
 
 
 class AtomType(ParametricPotential):
@@ -255,7 +255,9 @@ def validate_charge(cls, charge):
             warnings.warn(
                 UNIT_WARNING_STRING.format("Charges", "elementary charge")
             )
-            charge *= u.Unit("elementary_charge", registry=GMSO_UnitRegistry().reg_)
+            charge *= u.Unit(
+                "elementary_charge", registry=GMSO_UnitRegistry().reg_
+            )
         else:
             ensure_valid_dimensions(charge, u.elementary_charge)
 

diff --git a/gmso/core/forcefield.py b/gmso/core/forcefield.py
@@ -7,7 +7,6 @@
 from typing import Iterable
 
 from lxml import etree
-
 from pydantic import ValidationError
 
 from gmso.core.element import element_by_symbol
@@ -579,11 +578,11 @@ def xml_from_forcefield_utilities(cls, filename):
             loader = FoyerFFs()
             ff = loader.load(filename).to_gmso_ff()
             ff.units = {
-                    "energy": "kJ",
-                    "distance": "nm",
-                    "mass": "amu",
-                    "charge": "elementary_charge",
-                }
+                "energy": "kJ",
+                "distance": "nm",
+                "mass": "amu",
+                "charge": "elementary_charge",
+            }
         return ff
 
     def to_xml(self, filename, overwrite=False, backend="gmso"):

diff --git a/gmso/utils/units.py b/gmso/utils/units.py
@@ -82,7 +82,9 @@ def default_reg():
 
 def register_general_units(reg: u.UnitRegistry):
     """Register units that are generally useful to a basic unyt UnitSystem."""
-    elementary_charge_conversion = 1 * getattr(u.physical_constants, "elementary_charge").value
+    elementary_charge_conversion = (
+        1 * getattr(u.physical_constants, "elementary_charge").value
+    )
     charge_dim = u.dimensions.current_mks * u.dimensions.time
     reg.add(
         "elementary_charge",
@@ -111,7 +113,10 @@ def register_general_units(reg: u.UnitRegistry):
     )
     bohr_rad_dim = u.dimensions.length
     reg.add(
-        "a0", base_value=bohr_rad_conversion, dimensions=bohr_rad_dim, tex_repr=r"\rm{a0}"
+        "a0",
+        base_value=bohr_rad_conversion,
+        dimensions=bohr_rad_dim,
+        tex_repr=r"\rm{a0}",
     )  # bohr radius
 
     hartree_conversion = (