Skip to content

Commit d7081c9

Browse files
Vivek Miglanifacebook-github-bot
authored andcommitted
Fix Lime pyre fixme issues
Summary: Fixing unresolved pyre fixme issues in corresponding file Differential Revision: D76736867
1 parent 485c50d commit d7081c9

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

captum/attr/_core/lime.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -548,8 +548,7 @@ def generate_perturbation() -> (
548548

549549
return generate_perturbation
550550

551-
# pyre-fixme[24] Generic type `Callable` expects 2 type parameters.
552-
def attribute_future(self) -> Callable:
551+
def attribute_future(self) -> None:
553552
r"""
554553
This method is not implemented for LimeBase.
555554
"""
@@ -1116,8 +1115,7 @@ def attribute( # type: ignore
11161115
show_progress=show_progress,
11171116
)
11181117

1119-
# pyre-fixme[24] Generic type `Callable` expects 2 type parameters.
1120-
def attribute_future(self) -> Callable:
1118+
def attribute_future(self) -> None:
11211119
return super().attribute_future()
11221120

11231121
def _attribute_kwargs( # type: ignore

tests/attr/test_lime.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -482,7 +482,7 @@ def test_futures_not_implemented(self) -> None:
482482
)
483483
attributions = None
484484
with self.assertRaises(NotImplementedError):
485-
attributions = lime.attribute_future()
485+
attributions = lime.attribute_future() # type: ignore
486486
self.assertEqual(attributions, None)
487487

488488
# pyre-fixme[24]: Generic type `Callable` expects 2 type parameters.

0 commit comments

Comments
 (0)