Pass credential variables and make account optional

ncsa · Nov 8, 2022 · 64b4904 · 64b4904
1 parent 5980504
commit 64b4904
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 %}