From c265112359296ddb575368b7651a255c14762e28 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Sun, 4 Aug 2024 01:20:48 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 dpti/equi.py | 13 +++++++++++--
 dpti/hti.py  | 12 +++++++++---
 2 files changed, 20 insertions(+), 5 deletions(-)

diff --git a/dpti/equi.py b/dpti/equi.py
index 8ca703f..19e5ccc 100755
--- a/dpti/equi.py
+++ b/dpti/equi.py
@@ -35,7 +35,16 @@
 # def gen_equi_header(nsteps, prt_freq, dump_freq, temp, pres, tau_t, tau_p, mass_map, conf_file):
 # def gen_equi_header(nsteps, prt_freq, dump_freq, temp, pres, tau_t, tau_p, mass_map, conf_file):
 def gen_equi_header(
-    nsteps, thermo_freq, dump_freq, mass_map, temp, tau_t, tau_p, equi_conf, pres=None, custom_variables=None
+    nsteps,
+    thermo_freq,
+    dump_freq,
+    mass_map,
+    temp,
+    tau_t,
+    tau_p,
+    equi_conf,
+    pres=None,
+    custom_variables=None,
 ):
     ret = ""
     ret += "clear\n"
@@ -87,7 +96,7 @@ def gen_equi_force_field(model, if_meam=False, meam_model=None, append=None):
     if not if_meam:
         ret += "pair_style      deepmd %s" % model
         if append is not None:
-            ret += " "+append
+            ret += " " + append
         ret += "\n"
         ret += "pair_coeff * *\n"
     else:
diff --git a/dpti/hti.py b/dpti/hti.py
index f5d2fb4..466aa00 100755
--- a/dpti/hti.py
+++ b/dpti/hti.py
@@ -274,7 +274,9 @@ def _ff_spring(lamb, m_spring_k, var_spring):
     return ret
 
 
-def _ff_soft_lj(lamb, model, m_spring_k, step, sparam, if_meam=False, meam_model=None, append=None):
+def _ff_soft_lj(
+    lamb, model, m_spring_k, step, sparam, if_meam=False, meam_model=None, append=None
+):
     ret = ""
     ret += "# --------------------- FORCE FIELDS ---------------------\n"
     if step == "lj_on":
@@ -282,11 +284,15 @@ def _ff_soft_lj(lamb, model, m_spring_k, step, sparam, if_meam=False, meam_model
         var_spring = False
     elif step == "deep_on":
         # ret += _ff_meam_on(lamb, model, sparam)
-        ret += _ff_deep_on(lamb, model, sparam, if_meam=if_meam, meam_model=meam_model, append=append)
+        ret += _ff_deep_on(
+            lamb, model, sparam, if_meam=if_meam, meam_model=meam_model, append=append
+        )
         var_spring = False
     elif step == "spring_off":
         # ret += _ff_meam_lj_off(lamb, model, sparam)
-        ret += _ff_lj_off(lamb, model, sparam, if_meam=if_meam, meam_model=meam_model, append=append)
+        ret += _ff_lj_off(
+            lamb, model, sparam, if_meam=if_meam, meam_model=meam_model, append=append
+        )
         var_spring = True
     else:
         raise RuntimeError("unkown step", step)