Fix W&B publication setting (#585)

* Add W&B publication setting

* Switch to boolean

* Add wandb setting to the train action

* Run linter

* Change type name

* Make bool values lower case

taskcluster/configs/config.prod.yml

@@ -213,6 +213,9 @@ marian-args:
 # https://mozilla.github.io/firefox-translations-training/task-cluster.html#running-up-to-a-specific-step
 target-stage: all
 
+# Enable publication to Weights and Biases
+wandb-publication: true
+
 taskcluster:
   # After the parallel corpora are merged and de-duplicated, the combined file is
   # then split into an even number of chunks.

taskcluster/translations_taskgraph/actions/train.py

@@ -95,6 +95,11 @@ def validate_pretrained_models(params):
                 "default": defaults["target-stage"],
                 "enum": graph_config["valid-stages"],
             },
+            "wandb-publication": {
+                "type": "boolean",
+                "description": """Enable publication to Weights and Biases""",
+                "default": True,
+            },
             "experiment": {
                 "type": "object",
                 "default": defaults["experiment"],

tracking/translations_parser/cli/taskcluster.py

@@ -75,7 +75,7 @@ def get_args() -> argparse.Namespace:
     parser.add_argument(
         "--wandb-publication",
         action="store_true",
-        help="Trigger publication on Weight & Biases. Disabled by default. Can be set though env variable WANDB_PUBLICATION=true|false",
+        help="Trigger publication on Weight & Biases. Disabled by default. Can be set though env variable WANDB_PUBLICATION=True|False",
         default=os.environ.get("WANDB_PUBLICATION", "false").lower() == "true",
     )
     parser.add_argument(