From 1d818922e922b67ff0a5b5286583929d30780979 Mon Sep 17 00:00:00 2001 From: Ben Lindsay Date: Mon, 22 Jan 2018 10:32:16 -0500 Subject: [PATCH] smarter job_name assignment --- create_jobs/create_jobs.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/create_jobs/create_jobs.py b/create_jobs/create_jobs.py index fee5c16..29aa33b 100755 --- a/create_jobs/create_jobs.py +++ b/create_jobs/create_jobs.py @@ -44,7 +44,14 @@ def create_jobs(file_list=None, param_table=None, base_dir='.', # Create JOB_NAME column if not already there if not 'JOB_NAME' in param_df.columns: - param_df['JOB_NAME'] = param_df.index + job_name_col = None + for c in param_df.columns: + if np.all(~param_df[c].duplicated()): + job_name_col = c + if job_name_col is None: + param_df['JOB_NAME'] = param_df.index + else: + param_df['JOB_NAME'] = param_df[job_name_col] # Iterate over rows of dataframe, creating and submitting jobs param_dict_list = param_df.to_dict(orient='records')