From 0f755b861c3702b8516dc31eef0ad66528485778 Mon Sep 17 00:00:00 2001 From: Marc Glisse Date: Mon, 21 Nov 2022 01:11:30 +0100 Subject: Also update the test --- src/python/CMakeLists.txt | 5 +++++ src/python/test/test_perslay.py | 2 +- 2 files changed, 6 insertions(+), 1 deletion(-) (limited to 'src') diff --git a/src/python/CMakeLists.txt b/src/python/CMakeLists.txt index 344e988f..f6750a30 100644 --- a/src/python/CMakeLists.txt +++ b/src/python/CMakeLists.txt @@ -572,6 +572,11 @@ if(PYTHONINTERP_FOUND) add_gudhi_py_test(test_diff) endif() + # Perslay + if(TENSORFLOW_FOUND AND SKLEARN_FOUND) + add_gudhi_py_test(test_perslay) + endif() + # Betti curves if(SKLEARN_FOUND AND SCIPY_FOUND) add_gudhi_py_test(test_betti_curve_representations) diff --git a/src/python/test/test_perslay.py b/src/python/test/test_perslay.py index bf0d54af..06497712 100644 --- a/src/python/test/test_perslay.py +++ b/src/python/test/test_perslay.py @@ -1,7 +1,7 @@ import numpy as np import tensorflow as tf from sklearn.preprocessing import MinMaxScaler -from gudhi.tensorflow import * +from gudhi.tensorflow.perslay import * import gudhi.representations as gdr def test_gaussian_perslay(): -- cgit v1.2.3