Skip to content

Commit

Permalink
tests
Browse files Browse the repository at this point in the history
  • Loading branch information
JHoelli committed Nov 25, 2024
1 parent 9ff06eb commit 2bfd101
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/saliency/test_TSR.py
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,7 @@ def test_tsr_torch_explainer(tsr_torch_explainer,tsr):
exp= method.explain(x,labels=int(np.argmax(y[0])),TSR = tsr)
assert np.array(exp).shape == (1, X.shape[-1])

@pytest.mark.parametrize("tsr_tensorflow_explainer", ['GRAD','IG','GS','DLS','SG'],
@pytest.mark.parametrize("tsr_tensorflow_explainer", ['GRAD','IG','GS'],#,'DLS','SG'],
indirect=True)
@pytest.mark.parametrize('tsr',[True,False])
def test_tsr_tensorflow_explainer(tsr_tensorflow_explainer,tsr):
Expand Down

0 comments on commit 2bfd101

Please sign in to comment.