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

for more information, see https://pre-commit.ci
njzjz · Nov 22, 2024 · f4c9198 · f4c9198
1 parent e445372
commit f4c9198
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/source/api_cc/src/DeepPotPT.cc b/source/api_cc/src/DeepPotPT.cc
@@ -89,7 +89,10 @@ void DeepPotPT::init(const std::string& model,
   std::unordered_map<std::string, std::string> metadata = {{"type", ""}};
   module = torch::jit::load(model, device, metadata);
   module.eval();
-  const std::vector<std::string>& preserved_attrs = {"forward_lower", "has_message_passing", "get_rcut", "get_ntypes", "get_dim_fparam", "get_dim_aparam", "is_aparam_nall", "get_type_map"};
+  const std::vector<std::string>& preserved_attrs = {
+      "forward_lower",  "has_message_passing", "get_rcut",
+      "get_ntypes",     "get_dim_fparam",      "get_dim_aparam",
+      "is_aparam_nall", "get_type_map"};
   module = torch::jit::freeze(module, preserved_attrs);
   do_message_passing = module.run_method("has_message_passing").toBool();
   torch::jit::FusionStrategy strategy;

diff --git a/source/api_cc/src/DeepSpinPT.cc b/source/api_cc/src/DeepSpinPT.cc
@@ -89,7 +89,10 @@ void DeepSpinPT::init(const std::string& model,
   std::unordered_map<std::string, std::string> metadata = {{"type", ""}};
   module = torch::jit::load(model, device, metadata);
   module.eval();
-  const std::vector<std::string>& preserved_attrs = {"forward_lower", "has_message_passing", "get_rcut", "get_ntypes", "get_dim_fparam", "get_dim_aparam", "is_aparam_nall", "get_type_map"};
+  const std::vector<std::string>& preserved_attrs = {
+      "forward_lower",  "has_message_passing", "get_rcut",
+      "get_ntypes",     "get_dim_fparam",      "get_dim_aparam",
+      "is_aparam_nall", "get_type_map"};
   module = torch::jit::freeze(module, preserved_attrs);
   do_message_passing = module.run_method("has_message_passing").toBool();
   torch::jit::FusionStrategy strategy;