Merge pull request #2 from mailach/main

Change sbatch template to pass mlflow credential variables and make account information optional
ncsa · Nov 28, 2022 · b19eddf · b19eddf
2 parents 5980504 + 64b4904
commit b19eddf
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/mlflow_slurm/templates/sbatch_template.sh b/mlflow_slurm/templates/sbatch_template.sh
@@ -1,8 +1,10 @@
 #!/bin/bash
 #SBATCH --job-name=MLFlow{{ run_id }}
 #SBATCH --partition={{ config.partition }}
+{% if config.account %}
 #SBATCH --account={{ config.account }}
-#SBATCH --export=MLFLOW_TRACKING_URI,MLFLOW_S3_ENDPOINT_URL,AWS_SECRET_ACCESS_KEY,AWS_ACCESS_KEY_ID
+{% endif %}
+#SBATCH --export=MLFLOW_TRACKING_URI,MLFLOW_TRACKING_USERNAME,MLFLOW_TRACKING_PASSWORD,MLFLOW_S3_ENDPOINT_URL,AWS_SECRET_ACCESS_KEY,AWS_ACCESS_KEY_ID
 {% if config.gpus_per_node %}
 #SBATCH --gpus-per-node={{ config.gpus_per_node }}
 {% endif %}