From 6df157f3beba80da9fa28499bdfa390898166cb0 Mon Sep 17 00:00:00 2001
From: Thomas <purcell@fhi-berlin.mpg.de>
Date: Sun, 5 Sep 2021 22:56:33 +0200
Subject: [PATCH] Fix check models for systems with tests turned off

---
 tests/exec_test/classification_gen_proj/check_model.py | 2 +-
 tests/exec_test/gen_proj/check_model.py                | 2 +-
 tests/exec_test/log_reg_gen_proj/check_model.py        | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/exec_test/classification_gen_proj/check_model.py b/tests/exec_test/classification_gen_proj/check_model.py
index 5958c75a..c004eb5d 100644
--- a/tests/exec_test/classification_gen_proj/check_model.py
+++ b/tests/exec_test/classification_gen_proj/check_model.py
@@ -4,7 +4,7 @@ from pathlib import Path
 import numpy as np
 
 model = ModelClassifier(
-    str("models/train_dim_2_model_0.dat"), str("models/test_dim_2_model_0.dat")
+    str("models/train_dim_2_model_0.dat")
 )
 assert model.percent_error < 1e-7
 assert model.percent_test_error < 1e-7
diff --git a/tests/exec_test/gen_proj/check_model.py b/tests/exec_test/gen_proj/check_model.py
index 183a6c17..640d2b9a 100644
--- a/tests/exec_test/gen_proj/check_model.py
+++ b/tests/exec_test/gen_proj/check_model.py
@@ -4,7 +4,7 @@ from pathlib import Path
 import numpy as np
 
 model = ModelRegressor(
-    str("models/train_dim_2_model_0.dat"), str("models/test_dim_2_model_0.dat")
+    str("models/train_dim_2_model_0.dat")
 )
 assert model.rmse < 1e-4
 assert model.test_rmse < 1e-4
diff --git a/tests/exec_test/log_reg_gen_proj/check_model.py b/tests/exec_test/log_reg_gen_proj/check_model.py
index c5cdbe83..31153595 100644
--- a/tests/exec_test/log_reg_gen_proj/check_model.py
+++ b/tests/exec_test/log_reg_gen_proj/check_model.py
@@ -4,7 +4,7 @@ from pathlib import Path
 import numpy as np
 
 model = ModelLogRegressor(
-    str("models/train_dim_2_model_0.dat"), str("models/test_dim_2_model_0.dat")
+    str("models/train_dim_2_model_0.dat")
 )
 assert model.rmse < 1e-4
 assert model.test_rmse < 1e-4
-- 
GitLab