From 64b4904d9161752a94d427cf1ca3c01aebd1561f Mon Sep 17 00:00:00 2001
From: Alina Mailach <alina.mailach@informatik.uni-leipzig.de>
Date: Tue, 8 Nov 2022 10:06:21 +0100
Subject: [PATCH] Pass credential variables and make account optional

---
 mlflow_slurm/templates/sbatch_template.sh | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/mlflow_slurm/templates/sbatch_template.sh b/mlflow_slurm/templates/sbatch_template.sh
index c59b7bb..e1bc9cb 100644
--- 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 %}