Misc fixes

1. Fix linter issue in make_job_config
2. Fix typo in docker image creation script
3. Add importlib resources to ensure backwards compatibilty. Without that the tests seem to fail on ModuleNotFoundError.

Author: aahladc

docker/build_docker_images.sh

@@ -17,7 +17,7 @@ done
 
 # Artifact repostiory
 if [ "$PROJECT" = "mlcommons-algoperf" ]; then
-    ARTIFACT_REPO="europe-west-4-docker.pkg.dev/mlcommons-algoperf/algoperf-docker-repo"
+    ARTIFACT_REPO="europe-west4-docker.pkg.dev/mlcommons-algoperf/algoperf-docker-repo"
 else
     ARTIFACT_REPO="us-central1-docker.pkg.dev/training-algorithms-external/mlcommons-docker-repo"
 fi

pyproject.toml

@@ -46,7 +46,8 @@ dependencies = [
   "clu==0.0.12",
   "matplotlib>=3.9.2",
   "tabulate==0.9.0",
-  "wandb==0.21.0"
+  "wandb==0.21.0",
+  "importlib_resources"
 ]
 
 [build-system]

scoring/utils/slurm/make_job_config.py

@@ -67,7 +67,7 @@
   'librispeech_deepspeech': {'dataset': 'librispeech'},
   'criteo1tb': {'dataset': 'criteo1tb'},
   'librispeech_conformer': {'dataset': 'librispeech'},
-  'finewebedu_lm': {'dataset': 'fineweb_edu_10B'}
+  'finewebedu_lm': {'dataset': 'fineweb_edu_10B'},
 }
 
 RULESET_CONFIGS = {
@@ -92,7 +92,9 @@ def main(_):
 
   num_tuning_trials = FLAGS.num_tuning_trials
   if num_tuning_trials is None:
-    num_tuning_trials = RULESET_CONFIGS[FLAGS.tuning_ruleset]['num_tuning_trials']
+    num_tuning_trials = RULESET_CONFIGS[FLAGS.tuning_ruleset][
+      'num_tuning_trials'
+    ]
 
   key = jax.random.PRNGKey(FLAGS.seed)