Skip to content

Commit ddb237c

Browse files
committed
TEST add test checking that kernels(GP) cannot be uploaded
1 parent a9adb14 commit ddb237c

1 file changed

Lines changed: 11 additions & 2 deletions

File tree

tests/flows/test_sklearn.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,13 +3,15 @@
33
import unittest
44

55
import numpy as np
6+
import scipy.optimize
7+
import scipy.stats
68
import sklearn.base
79
import sklearn.datasets
8-
import scipy.stats
910
import sklearn.decomposition
1011
import sklearn.dummy
1112
import sklearn.ensemble
1213
import sklearn.feature_selection
14+
import sklearn.gaussian_process
1315
import sklearn.model_selection
1416
import sklearn.pipeline
1517
import sklearn.preprocessing
@@ -378,5 +380,12 @@ def test_hypothetical_parameter_values(self):
378380
self.assertEqual(deserialized.get_params(), model.get_params())
379381
self.assertIsNot(deserialized, model)
380382

381-
383+
def test_gaussian_process(self):
384+
opt = scipy.optimize.fmin_l_bfgs_b
385+
kernel = sklearn.gaussian_process.kernels.Matern()
386+
gp = sklearn.gaussian_process.GaussianProcessClassifier(
387+
kernel=kernel, optimizer=opt)
388+
self.assertRaisesRegexp(TypeError, "Matern\(length_scale=1, nu=1.5\), "
389+
"<class 'sklearn.gaussian_process.kernels.Matern'>",
390+
sklearn_to_flow, gp)
382391

0 commit comments

Comments
 (0)