Skip to content
Snippets Groups Projects
Commit 487b9b84 authored by Christopher Randolph Rhodes's avatar Christopher Randolph Rhodes
Browse files

Corrected JSON validation in test method

parent a62aa5a5
No related branches found
No related tags found
No related merge requests found
......@@ -16,7 +16,7 @@ class WorkflowRunRecord(BaseModel):
input_filepath: str
output_filepath: str
success: bool
timer_results: Dict[str, str]
timer_results: Dict[str, float]
def infer_image_to_image(fpi, model, where_output, **kwargs) -> dict:
# read image file into memory
......
......@@ -27,6 +27,10 @@ class TestGetSessionObject(unittest.TestCase):
sesh = Session()
di = WorkflowRunRecord(
model_id='test_model',
input_filepath='/test/input/directory',
output_filepath='/test/output/fi.le',
success=True,
timer_results={'start': 0.123},
)
sesh.record_workflow_run(di)
with open(sesh.manifest_json, 'r') as fh:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment