fix test build models

Author: floatingCatty

dptb/tests/test_build_model.py

@@ -46,8 +46,7 @@ def test_build_nnsk_from_scratch():
         "overlap": False,
         "seed": 3982377700
     }
-    statistics = None
-    model = build_model(None, model_options, common_options, statistics)
+    model = build_model(None, model_options, common_options)
 
     assert isinstance(model, NNSK)
     assert model.device == "cpu"
@@ -111,9 +110,8 @@ def test_build_model_MIX_from_scratch():
         "overlap": False,
         "seed": 3982377700
     }
-    statistics = None
 
-    model = build_model(None, model_options, common_options, statistics)
+    model = build_model(None, model_options, common_options)
 
     assert isinstance(model, MIX)
     assert model.name == "mix"

dptb/tests/test_trainer.py

@@ -47,7 +47,7 @@ def test_fromscratch_noref_noval(self):
         jdata = self.jdata
         train_datasets = self.train_datasets
         model = build_model(None, model_options=jdata["model_options"], 
-                        common_options=jdata["common_options"], statistics=train_datasets.E3statistics())
+                        common_options=jdata["common_options"])
         trainer = Trainer(
             train_options=jdata["train_options"],
             common_options=jdata["common_options"],
@@ -73,7 +73,7 @@ def test_fromscratch_ref_noval(self):
         reference_datasets = build_dataset(**jdata["data_options"]["reference"], **jdata["common_options"])
 
         model = build_model(None, model_options=jdata["model_options"], 
-                        common_options=jdata["common_options"], statistics=train_datasets.E3statistics())
+                        common_options=jdata["common_options"])
 
         trainer = Trainer(
             train_options=jdata["train_options"],
@@ -100,7 +100,7 @@ def test_fromscratch_noref_val(self):
         validation_datasets = build_dataset(**jdata["data_options"]["validation"], **jdata["common_options"])
 
         model = build_model(None, model_options=jdata["model_options"], 
-                        common_options=jdata["common_options"], statistics=train_datasets.E3statistics())
+                        common_options=jdata["common_options"])
 
         trainer = Trainer(
             train_options=jdata["train_options"],
@@ -126,7 +126,7 @@ def test_initmodel_noref_nval(self):
         checkpoint = f"{rootdir}/test_sktb/output/test_valence/checkpoint/nnsk.best.pth"
         run_options.update({"init_model": checkpoint, "restart": None})
         model = build_model(checkpoint, model_options=jdata["model_options"], 
-                            common_options=jdata["common_options"], statistics=train_datasets.E3statistics())
+                            common_options=jdata["common_options"])
         trainer = Trainer(
             train_options=jdata["train_options"],
             common_options=jdata["common_options"],