From 29f45db2511aea5eea9b2a41e10e9781a7eb281a Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Mon, 30 Oct 2023 16:58:24 +0100
Subject: [PATCH] Rolled back utility module
---
model_server/util.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/model_server/util.py b/model_server/util.py
index 8bc071ab..1cc33753 100644
--- a/model_server/util.py
+++ b/model_server/util.py
@@ -1,10 +1,12 @@
from pathlib import Path
import re
from time import localtime, strftime
+from typing import List
import pandas as pd
from model_server.accessors import InMemoryDataAccessor, write_accessor_data_to_file
+from model_server.models import Model
def autonumber_new_directory(where: str, prefix: str) -> str:
"""
@@ -75,6 +77,7 @@ def loop_workflow(
files: list,
output_folder_path: str,
workflow_func: callable,
+ models: List[Model],
params: dict,
export_batch_csvs: bool = True,
write_intermediate_products: bool = True,
@@ -95,6 +98,7 @@ def loop_workflow(
export_kwargs = {
'input_file_path': ff,
'output_folder_path': output_folder_path,
+ 'models': models,
**params,
}
--
GitLab