From c47ad73e46bbaad7b2b003ecd890dc0e8f537172 Mon Sep 17 00:00:00 2001
From: shixian.shi <shixian.shi@alibaba-inc.com>
Date: 星期一, 29 一月 2024 15:49:45 +0800
Subject: [PATCH] fix device bug

---
 setup.py |    4 ++--
 1 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index f7e6ee6..561dea2 100644
--- a/setup.py
+++ b/setup.py
@@ -38,14 +38,14 @@
         # "protobuf",
         "tqdm",
         "hdbscan",
-        "umap",
+        "umap_learn",
         "jaconv",
         "hydra-core>=1.3.2",
     ],
     # train: The modules invoked when training only.
     "train": [
         "editdistance",
-        "wandb",
+        "tensorboardX",
     ],
     # all: The modules should be optionally installled due to some reason.
     #      Please consider moving them to "install" occasionally

--
Gitblit v1.9.1