From d9c808a8df49a747b484993f7b69dc9c4936992b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=8C=97=E5=BF=B5?= <lzr265946@alibaba-inc.com>
Date: Thu, 12 Oct 2023 17:13:20 +0800
Subject: [PATCH] update setup.py

---
 setup.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/setup.py b/setup.py
index ee1da1d64..069e39408 100644
--- a/setup.py
+++ b/setup.py
@@ -42,6 +42,7 @@ requirements = {
         "protobuf",
         "tqdm",
         "hdbscan",
+        "umap",
     ],
     # train: The modules invoked when training only.
     "train": [