chore: import umap only when needed

AFAgarap · Jun 6, 2021 · 50c658f · 50c658f
1 parent b3788a8
commit 50c658f
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/pt_datasets/__init__.py b/pt_datasets/__init__.py
@@ -36,7 +36,7 @@
 )
 
 __author__ = "Abien Fred Agarap"
-__version__ = "0.14.3"
+__version__ = "0.14.4"
 
 
 def list_datasets() -> str:

diff --git a/pt_datasets/encode_features.py b/pt_datasets/encode_features.py
@@ -17,7 +17,6 @@
 import MulticoreTSNE
 import numpy as np
 from sklearn.decomposition import PCA
-from umap import UMAP
 
 __author__ = "Abien Fred Agarap"
 
@@ -65,6 +64,8 @@ def encode_features(
             n_jobs=4, random_state=seed, n_components=dim
         )
     elif encoder == "umap":
+        from umap import UMAP
+
         encoder = UMAP(n_components=dim, random_state=seed)
     encoded_features = encoder.fit_transform(features)
     return encoded_features
diff --git a/setup.py b/setup.py
@@ -64,7 +64,7 @@ def _download_yelp():
 
 setup(
     name="pt-datasets",
-    version="0.14.3",
+    version="0.14.4",
     packages=["pt_datasets"],
     url="https://github.com/AFAgarap/pt-datasets",
     license="AGPL-3.0 License",