Make the config related changes

georgia-tech-db · Oct 24, 2023 · 81ab7a3 · 81ab7a3
1 parent 0ab5ff8
commit 81ab7a3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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(