From 81ab7a37e5429cce57995279ab34319a5dd06de8 Mon Sep 17 00:00:00 2001 From: Jineet Desai Date: Mon, 23 Oct 2023 22:54:04 -0400 Subject: [PATCH] Make the config related changes --- evadb/executor/create_function_executor.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/evadb/executor/create_function_executor.py b/evadb/executor/create_function_executor.py index 9f5d46e8be..52a3f17c9f 100644 --- a/evadb/executor/create_function_executor.py +++ b/evadb/executor/create_function_executor.py @@ -245,7 +245,8 @@ def handle_xgboost_function(self): dataframe=aggregated_batch.frames, label=arg_map["predict"], **settings ) model_path = os.path.join( - self.db.config.get_value("storage", "model_dir"), self.node.name + self.db.catalog().get_configuration_catalog_value("model_dir"), + self.node.name, ) pickle.dump(model, open(model_path, "wb")) self.node.metadata.append(