From 7ecfd161d3ca951282a0093e12b4ce93ca3a15c2 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Sun, 6 Aug 2023 00:39:39 +0200
Subject: [PATCH] [tests] Remove traces of signs-to-tb "model"
---
tests/test_cli.py | 12 ++++++------
1 file changed, 6 insertions(+), 6 deletions(-)
diff --git a/tests/test_cli.py b/tests/test_cli.py
index 3d675032..888794b4 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -513,17 +513,17 @@ def test_compare_pasa_montgomery(temporary_basedir):
@pytest.mark.skip(reason="Test need to be updated")
@pytest.mark.skip_if_rc_var_not_set("datadir.montgomery")
-def test_train_signstotb_montgomery_rs(temporary_basedir, datadir):
+def test_train_mlp_montgomery_rs(temporary_basedir, datadir):
from ptbench.scripts.train import train
runner = CliRunner()
with stdout_logging() as buf:
- output_folder = str(temporary_basedir / "results/signstotb")
+ output_folder = str(temporary_basedir / "results/mlp")
result = runner.invoke(
train,
[
- "signs_to_tb",
+ "mlp",
"montgomery_rs",
"-vv",
"--epochs=1",
@@ -567,7 +567,7 @@ def test_train_signstotb_montgomery_rs(temporary_basedir, datadir):
@pytest.mark.skip(reason="Test need to be updated")
@pytest.mark.skip_if_rc_var_not_set("datadir.montgomery")
-def test_predict_signstotb_montgomery_rs(temporary_basedir, datadir):
+def test_predict_mlp_montgomery_rs(temporary_basedir, datadir):
from ptbench.scripts.predict import predict
runner = CliRunner()
@@ -577,12 +577,12 @@ def test_predict_signstotb_montgomery_rs(temporary_basedir, datadir):
result = runner.invoke(
predict,
[
- "signs_to_tb",
+ "mlp",
"montgomery_rs",
"-vv",
"--batch-size=1",
"--relevance-analysis",
- f"--weight={str(datadir / 'lfs' / 'models' / 'signstotb.ckpt')}",
+ f"--weight={str(datadir / 'lfs' / 'models' / 'mlp.ckpt')}",
f"--output-folder={output_folder}",
],
)
--
GitLab