From cb31ede9c11d1ca7514f75f9abfbb4c5266043b5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=88=98=E6=A3=92=E6=A3=92?= <liubangbang@fuzhi.ai>
Date: Tue, 19 Dec 2023 17:58:22 +0800
Subject: [PATCH] add udf in ML_MODULE_MAP.

---
 metagpt/prompts/ml_engineer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/metagpt/prompts/ml_engineer.py b/metagpt/prompts/ml_engineer.py
index 33eb9c40c..cca9649b3 100644
--- a/metagpt/prompts/ml_engineer.py
+++ b/metagpt/prompts/ml_engineer.py
@@ -301,6 +301,7 @@ ML_SPECIFIC_PROMPT = {
 ML_MODULE_MAP = {
     "data_preprocess": "metagpt.tools.functions.libs.data_preprocess",
     "feature_engineering": "metagpt.tools.functions.libs.feature_engineering",
+    "udf": "metagpt.tools.functions.libs.udf",
 }
 
 STRUCTURAL_CONTEXT = """