Skip to content

Commit 525e811

Browse files
Vivek Miglanifacebook-github-bot
authored andcommitted
Fix utils/common pyre fixme issues (#1588)
Summary: Pull Request resolved: #1588 Fixing unresolved pyre fixme issues in corresponding file Reviewed By: cyrjano Differential Revision: D76738238 fbshipit-source-id: 47c4dadad27360681aee33f087f5f343c091c250
1 parent bd6a881 commit 525e811

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

captum/_utils/common.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ def _is_tuple(inputs: Tensor) -> Literal[False]: ...
9191

9292
@typing.overload
9393
def _is_tuple(
94-
inputs: TensorOrTupleOfTensorsGeneric, # type: ignore
94+
inputs: Union[Tensor, Tuple[Tensor, ...]],
9595
) -> bool: ...
9696

9797

@@ -373,8 +373,6 @@ def _expand_target(
373373
def _expand_feature_mask(
374374
feature_mask: Union[Tensor, Tuple[Tensor, ...]], n_samples: int
375375
) -> Tuple[Tensor, ...]:
376-
# pyre-fixme[6]: For 1st argument expected `Tensor` but got `Union[Tensor,
377-
# typing.Tuple[Tensor, ...]]`.
378376
is_feature_mask_tuple = _is_tuple(feature_mask)
379377
feature_mask = _format_tensor_into_tuples(feature_mask)
380378
feature_mask_new = tuple(

0 commit comments

Comments
 (0)