File tree 2 files changed +3
-3
lines changed
alibi_detect/saving/tests
2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change 26
26
from alibi_detect .cd .pytorch import HiddenOutput as HiddenOutput_pt
27
27
from alibi_detect .cd .tensorflow import HiddenOutput as HiddenOutput_tf
28
28
from alibi_detect .utils .frameworks import has_keops
29
- if has_keops :
29
+ if has_keops : # pykeops only installed in Linux CI
30
30
from alibi_detect .utils .keops .kernels import GaussianRBF as GaussianRBF_ke
31
31
from alibi_detect .utils .keops .kernels import DeepKernel as DeepKernel_ke
32
32
Original file line number Diff line number Diff line change 46
46
from alibi_detect .utils .pytorch .kernels import DeepKernel as DeepKernel_pt
47
47
from alibi_detect .utils .tensorflow .kernels import DeepKernel as DeepKernel_tf
48
48
from alibi_detect .utils .frameworks import has_keops
49
- if has_keops :
49
+ if has_keops : # pykeops only installed in Linux CI
50
50
from pykeops .torch import LazyTensor
51
51
from alibi_detect .utils .keops .kernels import DeepKernel as DeepKernel_ke
52
52
56
56
# TODO: We currently parametrize encoder_model etc (in models.py) with backend, so the same flavour of
57
57
# preprocessing is used as the detector backend. In the future we could decouple this in tests.
58
58
backends = ['tensorflow' , 'pytorch' , 'sklearn' ]
59
- if has_keops : # keops currently only installed during linux CI
59
+ if has_keops : # pykeops only installed in Linux CI
60
60
backends .append ('keops' )
61
61
backend = param_fixture ("backend" , backends )
62
62
P_VAL = 0.05
You can’t perform that action at this time.
0 commit comments