fix test_model

SDCCA · Aug 21, 2024 · 2b2a3ed · 2b2a3ed
1 parent ff34d60
commit 2b2a3ed
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dgl_ptm/tests/test_model.py b/dgl_ptm/tests/test_model.py
@@ -173,7 +173,7 @@ def test_set_model_parameters(self):
 
     def test_set_model_parameters_with_file(self, config_file):
         model = dgl_ptm.PovertyTrapModel(model_identifier='initialize_model', root_path='test_models')
-        model.set_model_parameters(parameterFilePath=config_file)
+        model.set_model_parameters(parameter_file_path=config_file)
 
         assert model._model_identifier == 'initialize_model'
         assert model.steering_parameters['del_method'] == 'probability'
@@ -192,7 +192,7 @@ def test_set_model_parameters_with_kwargs(self):
     def test_set_model_parameters_with_file_and_kwargs(self, config_file):
         model = dgl_ptm.PovertyTrapModel(model_identifier='initialize_model', root_path='test_models')
         model.set_model_parameters(
-            parameterFilePath=config_file,
+            parameter_file_path=config_file,
             steering_parameters={'del_method': 'probability','del_threshold': 0.06}
             )