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

This reverts commit f4c9198.
njzjz · Nov 22, 2024 · a4fc9c8 · a4fc9c8
1 parent f4c9198
commit a4fc9c8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 8 deletions.
diff --git a/source/api_cc/src/DeepPotPT.cc b/source/api_cc/src/DeepPotPT.cc
@@ -89,10 +89,7 @@ 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,10 +89,7 @@ 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;