Skip to content

Commit b881553

Browse files
cicichen01facebook-github-bot
authored andcommitted
Move helper functions for influence tests to be under helpers directory (#1260)
Summary: As titled. This changes helps keeping the tests to be an exact mapped of the implementations. A future follow up could be a consolidation of helpers. (there could be some helpers universal for both attribution and influence) Differential Revision: D55270852
1 parent 0f6e47b commit b881553

13 files changed

+11
-11
lines changed
File renamed without changes.
File renamed without changes.

tests/influence/_core/test_arnoldi_influence.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
from parameterized import parameterized
2020
from tests.helpers import BaseTest
2121
from tests.helpers.basic import assertTensorAlmostEqual
22-
from tests.influence._utils.common import (
22+
from tests.helpers.influence.common import (
2323
_format_batch_into_tuple,
2424
build_test_name_func,
2525
DataInfluenceConstructor,

tests/influence/_core/test_dataloader.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
from parameterized import parameterized
1111
from tests.helpers import BaseTest
1212
from tests.helpers.basic import assertTensorAlmostEqual
13-
from tests.influence._utils.common import (
13+
from tests.helpers.influence.common import (
1414
_format_batch_into_tuple,
1515
build_test_name_func,
1616
DataInfluenceConstructor,

tests/influence/_core/test_naive_influence.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
from parameterized import parameterized
1616
from tests.helpers import BaseTest
1717
from tests.helpers.basic import assertTensorAlmostEqual, assertTensorTuplesAlmostEqual
18-
from tests.influence._utils.common import (
18+
from tests.helpers.influence.common import (
1919
_format_batch_into_tuple,
2020
build_test_name_func,
2121
DataInfluenceConstructor,

tests/influence/_core/test_tracin_aggregate_influence.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
from captum.influence._core.tracincp import TracInCP
1010
from parameterized import parameterized
1111
from tests.helpers.basic import assertTensorAlmostEqual, BaseTest
12-
from tests.influence._utils.common import (
12+
from tests.helpers.influence.common import (
1313
build_test_name_func,
1414
DataInfluenceConstructor,
1515
get_random_model_and_data,

tests/influence/_core/test_tracin_intermediate_quantities.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
from parameterized import parameterized
1515
from tests.helpers import BaseTest
1616
from tests.helpers.basic import assertTensorAlmostEqual
17-
from tests.influence._utils.common import (
17+
from tests.helpers.influence.common import (
1818
_format_batch_into_tuple,
1919
build_test_name_func,
2020
DataInfluenceConstructor,

tests/influence/_core/test_tracin_k_most_influential.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from parameterized import parameterized
99
from tests.helpers import BaseTest
1010
from tests.helpers.basic import assertTensorAlmostEqual
11-
from tests.influence._utils.common import (
11+
from tests.helpers.influence.common import (
1212
_format_batch_into_tuple,
1313
build_test_name_func,
1414
DataInfluenceConstructor,

tests/influence/_core/test_tracin_regression.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
from parameterized import parameterized
1515
from tests.helpers import BaseTest
1616
from tests.helpers.basic import assertTensorAlmostEqual
17-
from tests.influence._utils.common import (
17+
from tests.helpers.influence.common import (
1818
_isSorted,
1919
_wrap_model_in_dataparallel,
2020
build_test_name_func,

tests/influence/_core/test_tracin_self_influence.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
from parameterized import parameterized
1111
from tests.helpers import BaseTest
1212
from tests.helpers.basic import assertTensorAlmostEqual
13-
from tests.influence._utils.common import (
13+
from tests.helpers.influence.common import (
1414
_format_batch_into_tuple,
1515
build_test_name_func,
1616
DataInfluenceConstructor,

0 commit comments

Comments
 (0)