Add basic helloworld script

jobs/slurm/helloworld/environment.yml

@@ -0,0 +1,9 @@
+name: helloworld
+channels:
+  - conda-forge
+dependencies:
+  - python=3.12
+  - pip
+  - pip:
+    - wandb
+    - weave

jobs/slurm/helloworld/helloworld.py

@@ -0,0 +1,12 @@
+import wandb
+from wandb.sdk import launch
+import os
+import time
+
+name = f"{os.environ["SLURM_JOB_NAME"]} ({os.environ["SLURM_JOB_ID"]})"
+wandb.init(project="helloworld-slurm", name=name, group=f"slurm-{os.environ["SLURM_JOB_ID"]}", config={"hello": "default"})
+launch.manage_wandb_config(include=["hello"])
+
+print("wandb.config: ", dict(wandb.config))
+time.sleep(30)
+print("Goodbye world!")

jobs/slurm/helloworld/helloworld.sh

@@ -0,0 +1,21 @@
+#!/bin/bash
+
+#SBATCH --job-name=example1
+#SBATCH --nodes=1
+#SBATCH --ntasks-per-node=1
+#SBATCH --output=slurm-%x-%j.out
+
+source ~/.bashrc
+
+set -x -e
+
+echo "START TIME: $(date)"
+
+LOG_PATH="helloworld.log"
+
+conda activate helloworld
+export CMD="python helloworld.py"
+
+srun --jobid $SLURM_JOBID bash -c "$CMD" 2>&1 | tee -a $LOG_PATH
+
+echo "END TIME: $(date)"