Update tests

mozilla · May 22, 2024 · 044d93b · 044d93b
1 parent 468f6fe
commit 044d93b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/test_tracking_cli.py b/tests/test_tracking_cli.py
@@ -123,9 +123,9 @@ def test_experiments_marian_1_10(wandb_mock, getargs_mock, caplog, samples_dir,
             ),
             (logging.INFO, "Found 2 quantized metrics from speed folder"),
             (logging.INFO, "Found 16 metrics from task logs"),
-            (logging.INFO, "Creating missing run backward with associated metrics"),
+            (logging.INFO, "Creating missing run backwards with associated metrics"),
             (logging.INFO, "Creating missing run quantized with associated metrics"),
-            (logging.INFO, "Creating missing run student-finetuned with associated metrics"),
+            (logging.INFO, "Creating missing run student-finetune with associated metrics"),
             (logging.INFO, "Creating missing run teacher-base-0 with associated metrics"),
             (logging.INFO, "Creating missing run teacher-base-1 with associated metrics"),
             (logging.INFO, "Creating missing run teacher-ensemble with associated metrics"),

diff --git a/tests/test_tracking_utils.py b/tests/test_tracking_utils.py
@@ -33,7 +33,7 @@
         ),
         (
             "eval_student-finetuned_flores_devtest",
-            ("student-finetuned", "flores", "devtest", None),
+            ("student-finetune", "flores", "devtest", None),
         ),
         (
             "eval_teacher-base0_flores_devtest",
@@ -65,7 +65,7 @@
         ),
         (
             "evaluate-backward-url-gcp_pytest-dataset_a0017e-en-ru",
-            ("backward", "url", "gcp_pytest-dataset_a0017e", None),
+            ("backwards", "url", "gcp_pytest-dataset_a0017e", None),
         ),
         (
             "train-teacher-ast-en-1",