From 2c5da59c23ef1758705357f6f223b2fee1706ee4 Mon Sep 17 00:00:00 2001
From: Xudong Sun <smilesun@users.noreply.github.com>
Date: Sat, 23 Sep 2023 23:54:44 +0200
Subject: [PATCH] Update exp_utils.py

---
 domainlab/compos/exp/exp_utils.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/domainlab/compos/exp/exp_utils.py b/domainlab/compos/exp/exp_utils.py
index 3e39a11a0..d423958f2 100644
--- a/domainlab/compos/exp/exp_utils.py
+++ b/domainlab/compos/exp/exp_utils.py
@@ -63,6 +63,9 @@ def mk_model_na(self, tag=None, dd_cut=19):
         model_name = "_".join(list4mname)
         if self.host.args.debug:
             model_name = "debug_" + model_name
+        slurm = os.environ.get('SLURM_JOB_ID')
+        if slurm:
+            model_name = model_name + '_' + slurm
         logger = Logger.get_logger()
         logger.info(f"model name: {model_name}")
         return model_name