Move helper functions for influence tests to be under helpers directo…

…ry (#1260) Summary: Pull Request resolved: #1260 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) Reviewed By: cyrjano Differential Revision: D55270852 fbshipit-source-id: 68f06af70b2e8438282842176e4bba74c2442ffb
pytorch · Mar 25, 2024 · c6e9c22 · c6e9c22
1 parent 724e9a1
commit c6e9c22
Show file tree

Hide file tree

Showing 13 changed files with 11 additions and 11 deletions.
diff --git a/tests/influence/_utils/__init__.py → tests/helpers/influence/__init__.py b/tests/influence/_utils/__init__.py → tests/helpers/influence/__init__.py
diff --git a/tests/influence/_utils/common.py → tests/helpers/influence/common.py b/tests/influence/_utils/common.py → tests/helpers/influence/common.py
diff --git a/tests/influence/_core/test_arnoldi_influence.py b/tests/influence/_core/test_arnoldi_influence.py
@@ -19,7 +19,7 @@
 from parameterized import parameterized
 from tests.helpers import BaseTest
 from tests.helpers.basic import assertTensorAlmostEqual
-from tests.influence._utils.common import (
+from tests.helpers.influence.common import (
     _format_batch_into_tuple,
     build_test_name_func,
     DataInfluenceConstructor,

diff --git a/tests/influence/_core/test_dataloader.py b/tests/influence/_core/test_dataloader.py
@@ -10,7 +10,7 @@
 from parameterized import parameterized
 from tests.helpers import BaseTest
 from tests.helpers.basic import assertTensorAlmostEqual
-from tests.influence._utils.common import (
+from tests.helpers.influence.common import (
     _format_batch_into_tuple,
     build_test_name_func,
     DataInfluenceConstructor,

diff --git a/tests/influence/_core/test_naive_influence.py b/tests/influence/_core/test_naive_influence.py
@@ -15,7 +15,7 @@
 from parameterized import parameterized
 from tests.helpers import BaseTest
 from tests.helpers.basic import assertTensorAlmostEqual, assertTensorTuplesAlmostEqual
-from tests.influence._utils.common import (
+from tests.helpers.influence.common import (
     _format_batch_into_tuple,
     build_test_name_func,
     DataInfluenceConstructor,

diff --git a/tests/influence/_core/test_tracin_aggregate_influence.py b/tests/influence/_core/test_tracin_aggregate_influence.py
@@ -9,7 +9,7 @@
 from captum.influence._core.tracincp import TracInCP
 from parameterized import parameterized
 from tests.helpers.basic import assertTensorAlmostEqual, BaseTest
-from tests.influence._utils.common import (
+from tests.helpers.influence.common import (
     build_test_name_func,
     DataInfluenceConstructor,
     get_random_model_and_data,

diff --git a/tests/influence/_core/test_tracin_intermediate_quantities.py b/tests/influence/_core/test_tracin_intermediate_quantities.py
@@ -14,7 +14,7 @@
 from parameterized import parameterized
 from tests.helpers import BaseTest
 from tests.helpers.basic import assertTensorAlmostEqual
-from tests.influence._utils.common import (
+from tests.helpers.influence.common import (
     _format_batch_into_tuple,
     build_test_name_func,
     DataInfluenceConstructor,

diff --git a/tests/influence/_core/test_tracin_k_most_influential.py b/tests/influence/_core/test_tracin_k_most_influential.py
@@ -8,7 +8,7 @@
 from parameterized import parameterized
 from tests.helpers import BaseTest
 from tests.helpers.basic import assertTensorAlmostEqual
-from tests.influence._utils.common import (
+from tests.helpers.influence.common import (
     _format_batch_into_tuple,
     build_test_name_func,
     DataInfluenceConstructor,

diff --git a/tests/influence/_core/test_tracin_regression.py b/tests/influence/_core/test_tracin_regression.py
@@ -14,7 +14,7 @@
 from parameterized import parameterized
 from tests.helpers import BaseTest
 from tests.helpers.basic import assertTensorAlmostEqual
-from tests.influence._utils.common import (
+from tests.helpers.influence.common import (
     _isSorted,
     _wrap_model_in_dataparallel,
     build_test_name_func,

diff --git a/tests/influence/_core/test_tracin_self_influence.py b/tests/influence/_core/test_tracin_self_influence.py
@@ -10,7 +10,7 @@
 from parameterized import parameterized
 from tests.helpers import BaseTest
 from tests.helpers.basic import assertTensorAlmostEqual
-from tests.influence._utils.common import (
+from tests.helpers.influence.common import (
     _format_batch_into_tuple,
     build_test_name_func,
     DataInfluenceConstructor,

diff --git a/tests/influence/_core/test_tracin_show_progress.py b/tests/influence/_core/test_tracin_show_progress.py
@@ -8,7 +8,7 @@
 from captum.influence._core.tracincp_fast_rand_proj import TracInCPFast
 from parameterized import parameterized
 from tests.helpers import BaseTest
-from tests.influence._utils.common import (
+from tests.helpers.influence.common import (
     build_test_name_func,
     DataInfluenceConstructor,
     get_random_model_and_data,

diff --git a/tests/influence/_core/test_tracin_validation.py b/tests/influence/_core/test_tracin_validation.py
@@ -7,7 +7,7 @@
 
 from parameterized import parameterized
 from tests.helpers import BaseTest
-from tests.influence._utils.common import (
+from tests.helpers.influence.common import (
     build_test_name_func,
     DataInfluenceConstructor,
     get_random_model_and_data,

diff --git a/tests/influence/_core/test_tracin_xor.py b/tests/influence/_core/test_tracin_xor.py
@@ -10,7 +10,7 @@
 from parameterized import parameterized
 from tests.helpers import BaseTest
 from tests.helpers.basic import assertTensorAlmostEqual
-from tests.influence._utils.common import (
+from tests.helpers.influence.common import (
     _wrap_model_in_dataparallel,
     BasicLinearNet,
     BinaryDataset,