Moving tests.attr.helpers to captum.testing.attr.helpers (#1492)

Summary:
Pull Request resolved: #1492

This Diff is to move the tests.attr.helpers functions away from the tests namespace and move it inside the captum library.

Reviewed By: jjuncho

Differential Revision: D68445008

fbshipit-source-id: eeeeb1211f570c5f6867db3886cf18438f6c0334

Author: styusuf

tests/attr/helpers/__init__.py renamed to captum/testing/attr/helpers/__init__.py

@@ -8,6 +8,7 @@
 import torch
 from captum._utils.typing import BaselineType
 from captum.attr._core.layer.layer_conductance import LayerConductance
+from captum.testing.attr.helpers.conductance_reference import ConductanceReference
 from captum.testing.helpers.basic import (
     assertTensorAlmostEqual,
     assertTensorTuplesAlmostEqual,
@@ -19,7 +20,6 @@
     BasicModel_MultiLayer_MultiInput,
 )
 from packaging import version
-from tests.attr.helpers.conductance_reference import ConductanceReference
 from torch import Tensor
 from torch.nn import Module
 

tests/attr/helpers/attribution_delta_util.py renamed to captum/testing/attr/helpers/attribution_delta_util.py

@@ -9,6 +9,10 @@
 
 import torch
 from captum.attr._core.layer.layer_deep_lift import LayerDeepLift, LayerDeepLiftShap
+from captum.testing.attr.helpers.neuron_layer_testing_util import (
+    create_inps_and_base_for_deeplift_neuron_layer_testing,
+    create_inps_and_base_for_deepliftshap_neuron_layer_testing,
+)
 from captum.testing.helpers.basic import (
     assert_delta,
     assertTensorAlmostEqual,
@@ -24,10 +28,6 @@
     ReLULinearModel,
 )
 from packaging import version
-from tests.attr.helpers.neuron_layer_testing_util import (
-    create_inps_and_base_for_deeplift_neuron_layer_testing,
-    create_inps_and_base_for_deepliftshap_neuron_layer_testing,
-)
 from torch import Tensor
 
 

tests/attr/helpers/conductance_reference.py renamed to captum/testing/attr/helpers/conductance_reference.py

@@ -13,6 +13,7 @@
     LayerGradientShap,
     LayerInputBaselineXGradient,
 )
+from captum.testing.attr.helpers.attribution_delta_util import assert_attribution_delta
 from captum.testing.helpers.basic import (
     assertTensorAlmostEqual,
     assertTensorTuplesAlmostEqual,
@@ -24,7 +25,6 @@
 )
 from captum.testing.helpers.classification_models import SoftmaxModel
 from packaging import version
-from tests.attr.helpers.attribution_delta_util import assert_attribution_delta
 from torch import Tensor
 from torch.nn import Module