Merge pull request #409 from JaGeo/fix_storage_issue

Fix issue with data store
materialsproject · Aug 31, 2023 · 1ac9023 · 1ac9023
2 parents 94a4984 + ba0e98e
commit 1ac9023
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/jobflow/core/job.py b/src/jobflow/core/job.py
@@ -1321,6 +1321,7 @@ def prepare_replace(
         store_output_job.index = current_job.index + 1
         store_output_job.metadata = current_job.metadata
         store_output_job.output_schema = current_job.output_schema
+        store_output_job._kwargs = current_job._kwargs
         replace.add_jobs(store_output_job)
 
     elif isinstance(replace, Job):

diff --git a/tests/core/test_job.py b/tests/core/test_job.py
@@ -179,7 +179,7 @@ def replace_list_flow():
         assert j.hosts == [response.replace.uuid]
 
     # replace with flow with outputs
-    test_job = Job(replace_flow, metadata=metadata, output_schema="123")
+    test_job = Job(replace_flow, metadata=metadata, output_schema="123", data=True)
     # wrap the job in a Flow to check hosts
     test_flow = Flow([test_job])
     response = test_job.run(memory_jobstore)
@@ -188,6 +188,7 @@ def replace_list_flow():
     assert response.replace.jobs[-1].uuid == test_job.uuid
     assert response.replace.jobs[-1].metadata == metadata
     assert response.replace.jobs[-1].output_schema == "123"
+    assert response.replace.jobs[-1]._kwargs["data"]
     assert response.replace.output is not None
     for j in response.replace.jobs:
         assert j.hosts == [response.replace.uuid, test_flow.uuid]