1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
Author: Andreas Tille <tille@debian.org>
Last-Update: Thu, 09 Feb 2023 15:13:30 +0100
Description: Properly skip tests needing tensorflow
--- a/umap/tests/test_parametric_umap.py
+++ b/umap/tests/test_parametric_umap.py
@@ -10,7 +10,7 @@
import tensorflow as tf
IMPORT_TF = True
-except ImportError:
+except ModuleNotFoundError:
IMPORT_TF = False
else:
from umap.parametric_umap import ParametricUMAP, load_ParametricUMAP
|
