Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
Signed-off-by: Future-Outlier <[email protected]>
  • Loading branch information
Future-Outlier committed Feb 22, 2025
1 parent 6b70901 commit cf8eafa
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions plugins/flytekit-slurm/tests/test_slurm_fn_task.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ def test_slurm_task():
script=script_file
)
)
def plus_one(x: int) -> int:
def plus_one(x: int) -> int:
return x + 1

assert plus_one.task_config is not None
Expand All @@ -53,5 +53,3 @@ def plus_one(x: int) -> int:
assert retrieved_settings["ssh_config"] == {"host": "your-slurm-host", "username": "ubuntu"}
assert retrieved_settings["sbatch_conf"] == {"partition": "debug", "job-name": "tiny-slurm", "output": "/home/ubuntu/fn_task.log"}
assert retrieved_settings["script"] == script_file


0 comments on commit cf8eafa

Please sign in to comment.