Modify UT

Chengqian-Zhang · Oct 30, 2024 · 16bd291 · 16bd291
1 parent 63b347d
commit 16bd291
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/source/tests/pt/test_dp_test.py b/source/tests/pt/test_dp_test.py
@@ -152,6 +152,7 @@ def setUp(self):
         self.config["training"]["training_data"]["systems"] = data_file
         self.config["training"]["validation_data"]["systems"] = data_file
         self.config["model"] = deepcopy(model_property)
+        self.config["model"]["type_map"] = [self.config["model"]["type_map"][i] for i in [1,0,3,2]]
         self.input_json = "test_dp_test_property.json"
         with open(self.input_json, "w") as fp:
             json.dump(self.config, fp, indent=4)