We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bd6a881 commit 525e811Copy full SHA for 525e811
captum/_utils/common.py
@@ -91,7 +91,7 @@ def _is_tuple(inputs: Tensor) -> Literal[False]: ...
91
92
@typing.overload
93
def _is_tuple(
94
- inputs: TensorOrTupleOfTensorsGeneric, # type: ignore
+ inputs: Union[Tensor, Tuple[Tensor, ...]],
95
) -> bool: ...
96
97
@@ -373,8 +373,6 @@ def _expand_target(
373
def _expand_feature_mask(
374
feature_mask: Union[Tensor, Tuple[Tensor, ...]], n_samples: int
375
) -> Tuple[Tensor, ...]:
376
- # pyre-fixme[6]: For 1st argument expected `Tensor` but got `Union[Tensor,
377
- # typing.Tuple[Tensor, ...]]`.
378
is_feature_mask_tuple = _is_tuple(feature_mask)
379
feature_mask = _format_tensor_into_tuples(feature_mask)
380
feature_mask_new = tuple(
0 commit comments