Function assertTensorTuplesAlmostEqual

captum/testing/helpers/basic.py:73–90

(
    test: unittest.TestCase,
    # pyre-fixme[2]: Parameter must be annotated.
    actual,
    # pyre-fixme[2]: Parameter must be annotated.
    expected,
    delta: float = 0.0001,
    mode: str = "sum",
)

def assertTensorTuplesAlmostEqual(
    test: unittest.TestCase,
    # pyre-fixme[2]: Parameter must be annotated.
    actual,
    # pyre-fixme[2]: Parameter must be annotated.
    expected,
    delta: float = 0.0001,
    mode: str = "sum",
) -> None:
    if isinstance(expected, tuple):
        assert len(actual) == len(
            expected
        ), f"the length of actual {len(actual)} != expected {len(expected)}"

        for i in range(len(expected)):
            assertTensorAlmostEqual(test, actual[i], expected[i], delta, mode)
    else:
        assertTensorAlmostEqual(test, actual, expected, delta, mode)


def assertTupleOfListOfTensorsAlmostEqual(