feat: config= parameter for formatting job names

Author: YodaEmbedding

README.md

@@ -97,18 +97,13 @@ easy_slurm.submit_job(
 
 The job names can be formatted using a config dictionary:
 ```python
-job_name = easy_slurm.format.format_with_config(
-    "bs={hp.batch_size:04},lr={hp.lr:.1e}",
-    config={"hp": {"batch_size": 32, "lr": 1e-2}},
-)
-
 easy_slurm.submit_job(
-    job_dir="$HOME/jobs/{date:%Y-%m-%d}-{job_name}",
     sbatch_options={
-        "job-name": job_name,  # equals "bs=0032,lr=1.0e-02"
-        ...
+        "job-name": "bs={hp.batch_size:04},lr={hp.lr:.1e}",
+        # Equivalent to:
+        # "job-name": "bs=0032,lr=1.0e-02"
     },
-    ...
+    config={"hp": {"batch_size": 32, "lr": 1e-2}},
 )
 ```
 

easy_slurm/jobs.py

@@ -37,6 +37,7 @@ def submit_job(
     submit: bool = True,
     interactive: bool = False,
     resubmit_limit: int = 64,
+    config: dict[str, Any] = {},
 ) -> str:
     """Submits job.
 
@@ -87,12 +88,17 @@ def submit_job(
             Maximum number of times to auto-submit a job for "resume".
             (Not entirely unlike submitting a resume for a job.)
             Default is 64 resubmissions.
+        config (dict[str, Any]):
+            A dictionary of configuration values to use for formatting.
 
     Returns:
         Path to the newly created job directory.
     """
     job_name = sbatch_options.get("job-name", "untitled")
-    job_dir = _expand_path(format_with_config(job_dir, {"job_name": job_name}))
+    job_name = format_with_config(job_name, config)
+    job_dir = _expand_path(
+        format_with_config(job_dir, {**config, "job_name": job_name})
+    )
     create_job_dir(job_dir, src)
 
     _write_script(