Skip to content

Commit

Permalink
Add test for case where additional store is not known to jobflow-remote
Browse files Browse the repository at this point in the history
  • Loading branch information
ml-evs committed Feb 18, 2024
1 parent 1d3f6f5 commit ce2ced0
Showing 1 changed file with 14 additions and 1 deletion.
15 changes: 14 additions & 1 deletion src/jobflow_remote/testing/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,20 @@ def check_env_var() -> str:

@job(big_files="data")
def add_big(a: float, b: float):
"""Adds two numbers together and writes the answer to an artificially large file."""
"""Adds two numbers together and writes the answer to an artificially large file
which is stored in a pre-defined store."""
import pathlib

result = a + b
with open("file.txt", "w") as f:
f.writelines([f"{result}"] * int(1e5))
return Response({"data": pathlib.Path("file.txt"), "result": a + b})


@job(undefined_store="data")
def add_big_undefined_store(a: float, b: float):
"""Adds two numbers together and writes the answer to an artificially large file
which is attempted to be stored in a undefined store."""
import pathlib

result = a + b
Expand Down

0 comments on commit ce2ced0

Please sign in to comment.