diff --git a/plugins/flytekit-slurm/demo.md b/plugins/flytekit-slurm/demo.md index e5ce90e5d3..03d2bba33f 100644 --- 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": "", + "port": "", + "username": "", + "password": "", + }, srun_conf={ "partition": "debug", "job-name": "demo-slurm",