Skip to content

Commit 5bb7314

Browse files
author
Ashley Scillitoe
committed
Add comments re has_keops
1 parent dbe24a9 commit 5bb7314

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

alibi_detect/saving/tests/models.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
from alibi_detect.cd.pytorch import HiddenOutput as HiddenOutput_pt
2727
from alibi_detect.cd.tensorflow import HiddenOutput as HiddenOutput_tf
2828
from alibi_detect.utils.frameworks import has_keops
29-
if has_keops:
29+
if has_keops: # pykeops only installed in Linux CI
3030
from alibi_detect.utils.keops.kernels import GaussianRBF as GaussianRBF_ke
3131
from alibi_detect.utils.keops.kernels import DeepKernel as DeepKernel_ke
3232

alibi_detect/saving/tests/test_saving.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@
4646
from alibi_detect.utils.pytorch.kernels import DeepKernel as DeepKernel_pt
4747
from alibi_detect.utils.tensorflow.kernels import DeepKernel as DeepKernel_tf
4848
from alibi_detect.utils.frameworks import has_keops
49-
if has_keops:
49+
if has_keops: # pykeops only installed in Linux CI
5050
from pykeops.torch import LazyTensor
5151
from alibi_detect.utils.keops.kernels import DeepKernel as DeepKernel_ke
5252

@@ -56,7 +56,7 @@
5656
# TODO: We currently parametrize encoder_model etc (in models.py) with backend, so the same flavour of
5757
# preprocessing is used as the detector backend. In the future we could decouple this in tests.
5858
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
6060
backends.append('keops')
6161
backend = param_fixture("backend", backends)
6262
P_VAL = 0.05

0 commit comments

Comments
 (0)