fix import module dependency

北念

2023-10-19 a9600a123e727e26abe31408118456185769d525

 setup.py

@@ -42,6 +42,7 @@
        "protobuf",
        "tqdm",
        "hdbscan",
        "umap",
    ],
    # train: The modules invoked when training only.
    "train": [