Fix training and model key detection

mozilla · Sep 3, 2024 · 5c17599 · 5c17599
1 parent 7dc10d0
commit 5c17599
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/taskcluster/translations_taskgraph/parameters.py b/taskcluster/translations_taskgraph/parameters.py
@@ -39,7 +39,7 @@ def get_defaults(_) -> dict:
             "marian-args": {
                 "training-backward": {
                     "disp-freq": "2",
-                    "save-freq": "26",
+                    "save-freq": "25",
                     "valid-freq": "50",
                     "after": "50u",
                     "dim-vocabs": "1000 1000",

diff --git a/tracking/translations_parser/parser.py b/tracking/translations_parser/parser.py
@@ -276,9 +276,9 @@ def flatten(vals):
 
         # Handle Marian model and training YAML configuration files (called as --config or -c)
         for path in args.get("config", args["c"]):
-            if path.endswith("train.yml"):
+            if path.startswith("configs/training"):
                 key = "training"
-            elif path.endswith(".yml"):
+            elif path.startswith("configs/model"):
                 key = "model"
             else:
                 continue