docs: Include ssh config in demo example

Signed-off-by: JiaWei Jiang <[email protected]>
flyteorg · Dec 31, 2024 · 9e6d8a6 · 9e6d8a6
1 parent 8229418
commit 9e6d8a6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/plugins/flytekit-slurm/demo.md b/plugins/flytekit-slurm/demo.md
@@ -84,6 +84,12 @@ from flytekitplugins.slurm import Slurm, SlurmTask
 
 @task(
     task_config=Slurm(
+        ssh_conf={
+            "host": "<ssh_host>",
+            "port": "<ssh_port>",
+            "username": "<ssh_username>",
+            "password": "<ssh_password>",
+        },
         srun_conf={
             "partition": "debug",
             "job-name": "demo-slurm",