Skip to content

Commit 448078b

Browse files
committed
Add "needs" field to Job resource
1 parent 42aa608 commit 448078b

File tree

3 files changed

+9
-1
lines changed

3 files changed

+9
-1
lines changed

metafold/jobs.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,7 @@ class Job:
7777
error: Optional[str] = None
7878
inputs: IO = field(converter=lambda v: v if isinstance(v, IO) else IO.from_dict(v))
7979
outputs: IO = field(converter=lambda v: v if isinstance(v, IO) else IO.from_dict(v))
80+
needs: list[str]
8081
# NOTE(ryan): Deprecated
8182
assets: Optional[list[Asset]] = field(
8283
converter=lambda v: optional(_assets)(v), default=None)

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
44

55
[project]
66
name = "metafold"
7-
version = "0.8.dev0"
7+
version = "0.8.dev1"
88
authors = [
99
{name = "Metafold 3D", email = "info@metafold3d.com"},
1010
]

tests/test_jobs.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@
5050
"outputs": {
5151
"params": None,
5252
},
53+
"needs": [],
5354
"parameters": default_params,
5455
"meta": None,
5556
},
@@ -68,6 +69,7 @@
6869
"outputs": {
6970
"params": None,
7071
},
72+
"needs": [],
7173
"parameters": default_params,
7274
"meta": None,
7375
},
@@ -86,6 +88,7 @@
8688
"outputs": {
8789
"params": None,
8890
},
91+
"needs": [],
8992
"parameters": default_params,
9093
"meta": None,
9194
},
@@ -109,6 +112,7 @@
109112
"outputs": {
110113
"params": None,
111114
},
115+
"needs": [],
112116
"assets": [],
113117
"parameters": {
114118
"foo": "1",
@@ -224,6 +228,7 @@ def test_get_job(client):
224228
error=None,
225229
inputs=IO(params=default_params),
226230
outputs=IO(),
231+
needs=[],
227232
assets=[asset_obj],
228233
parameters=default_params,
229234
meta=None,
@@ -247,6 +252,7 @@ def test_run_job(client):
247252
finished=default_dt,
248253
inputs=IO(params=params),
249254
outputs=IO(),
255+
needs=[],
250256
assets=[asset_obj],
251257
parameters=params,
252258
meta=None,
@@ -273,6 +279,7 @@ def test_poll_job(client):
273279
finished=default_dt,
274280
inputs=IO(params=params),
275281
outputs=IO(),
282+
needs=[],
276283
assets=[asset_obj],
277284
parameters=params,
278285
meta=None,

0 commit comments

Comments
 (0)