Skip to content

Commit 8b9983d

Browse files
yucufacebook-github-bot
yucu
authored and
facebook-github-bot
committed
Remove some linter override (#1270)
Summary: Pull Request resolved: #1270 As title Remove `# noqa` linter override from captum code Reviewed By: vivekmig Differential Revision: D55930958 fbshipit-source-id: 4350a3ea3de2cf60c986f0e42c37acfead3b5f03
1 parent c7a4e60 commit 8b9983d

File tree

10 files changed

+109
-90
lines changed

10 files changed

+109
-90
lines changed

‎captum/__init__.py‎

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,12 @@
11
#!/usr/bin/env python3
2-
import captum.attr as attr # noqa
3-
import captum.concept as concept # noqa
4-
import captum.influence as influence # noqa
5-
import captum.log as log # noqa
6-
import captum.metrics as metrics # noqa
7-
import captum.robust as robust # noqa
2+
import captum.attr as attr
3+
import captum.concept as concept
4+
import captum.influence as influence
5+
import captum.log as log
6+
import captum.metrics as metrics
7+
import captum.robust as robust
88

99

1010
__version__ = "0.7.0"
11+
12+
__all__ = ["attr", "concept", "influence", "log", "metrics", "robust"]

‎captum/attr/__init__.py‎

Lines changed: 41 additions & 57 deletions
Original file line numberDiff line numberDiff line change
@@ -1,82 +1,65 @@
11
#!/usr/bin/env python3
2-
from captum.attr._core.dataloader_attr import DataLoaderAttribution # noqa
3-
from captum.attr._core.deep_lift import DeepLift, DeepLiftShap # noqa
4-
from captum.attr._core.feature_ablation import FeatureAblation # noqa
5-
from captum.attr._core.feature_permutation import FeaturePermutation # noqa
6-
from captum.attr._core.gradient_shap import GradientShap # noqa
7-
from captum.attr._core.guided_backprop_deconvnet import ( # noqa
8-
Deconvolution,
9-
GuidedBackprop,
10-
)
11-
from captum.attr._core.guided_grad_cam import GuidedGradCam # noqa
12-
from captum.attr._core.input_x_gradient import InputXGradient # noqa
13-
from captum.attr._core.integrated_gradients import IntegratedGradients # noqa
14-
from captum.attr._core.kernel_shap import KernelShap # noqa
15-
from captum.attr._core.layer.grad_cam import LayerGradCam # noqa
16-
from captum.attr._core.layer.internal_influence import InternalInfluence # noqa
17-
from captum.attr._core.layer.layer_activation import LayerActivation # noqa
18-
from captum.attr._core.layer.layer_conductance import LayerConductance # noqa
19-
from captum.attr._core.layer.layer_deep_lift import ( # noqa
20-
LayerDeepLift,
21-
LayerDeepLiftShap,
22-
)
23-
from captum.attr._core.layer.layer_feature_ablation import LayerFeatureAblation # noqa
24-
from captum.attr._core.layer.layer_feature_permutation import ( # noqa
25-
LayerFeaturePermutation,
26-
)
27-
from captum.attr._core.layer.layer_gradient_shap import LayerGradientShap # noqa
28-
from captum.attr._core.layer.layer_gradient_x_activation import ( # noqa
29-
LayerGradientXActivation,
30-
)
31-
from captum.attr._core.layer.layer_integrated_gradients import ( # noqa
32-
LayerIntegratedGradients,
33-
)
34-
from captum.attr._core.layer.layer_lrp import LayerLRP # noqa
35-
from captum.attr._core.lime import Lime, LimeBase # noqa
36-
from captum.attr._core.llm_attr import ( # noqa
2+
from captum.attr._core.dataloader_attr import DataLoaderAttribution
3+
from captum.attr._core.deep_lift import DeepLift, DeepLiftShap
4+
from captum.attr._core.feature_ablation import FeatureAblation
5+
from captum.attr._core.feature_permutation import FeaturePermutation
6+
from captum.attr._core.gradient_shap import GradientShap
7+
from captum.attr._core.guided_backprop_deconvnet import Deconvolution, GuidedBackprop
8+
from captum.attr._core.guided_grad_cam import GuidedGradCam
9+
from captum.attr._core.input_x_gradient import InputXGradient
10+
from captum.attr._core.integrated_gradients import IntegratedGradients
11+
from captum.attr._core.kernel_shap import KernelShap
12+
from captum.attr._core.layer.grad_cam import LayerGradCam
13+
from captum.attr._core.layer.internal_influence import InternalInfluence
14+
from captum.attr._core.layer.layer_activation import LayerActivation
15+
from captum.attr._core.layer.layer_conductance import LayerConductance
16+
from captum.attr._core.layer.layer_deep_lift import LayerDeepLift, LayerDeepLiftShap
17+
from captum.attr._core.layer.layer_feature_ablation import LayerFeatureAblation
18+
from captum.attr._core.layer.layer_feature_permutation import LayerFeaturePermutation
19+
from captum.attr._core.layer.layer_gradient_shap import LayerGradientShap
20+
from captum.attr._core.layer.layer_gradient_x_activation import LayerGradientXActivation
21+
from captum.attr._core.layer.layer_integrated_gradients import LayerIntegratedGradients
22+
from captum.attr._core.layer.layer_lrp import LayerLRP
23+
from captum.attr._core.lime import Lime, LimeBase
24+
from captum.attr._core.llm_attr import (
3725
LLMAttribution,
3826
LLMAttributionResult,
3927
LLMGradientAttribution,
4028
)
41-
from captum.attr._core.lrp import LRP # noqa
42-
from captum.attr._core.neuron.neuron_conductance import NeuronConductance # noqa
43-
from captum.attr._core.neuron.neuron_deep_lift import ( # noqa
44-
NeuronDeepLift,
45-
NeuronDeepLiftShap,
46-
)
47-
from captum.attr._core.neuron.neuron_feature_ablation import ( # noqa
48-
NeuronFeatureAblation,
49-
)
50-
from captum.attr._core.neuron.neuron_gradient import NeuronGradient # noqa
51-
from captum.attr._core.neuron.neuron_gradient_shap import NeuronGradientShap # noqa
52-
from captum.attr._core.neuron.neuron_guided_backprop_deconvnet import ( # noqa
29+
from captum.attr._core.lrp import LRP
30+
from captum.attr._core.neuron.neuron_conductance import NeuronConductance
31+
from captum.attr._core.neuron.neuron_deep_lift import NeuronDeepLift, NeuronDeepLiftShap
32+
from captum.attr._core.neuron.neuron_feature_ablation import NeuronFeatureAblation
33+
from captum.attr._core.neuron.neuron_gradient import NeuronGradient
34+
from captum.attr._core.neuron.neuron_gradient_shap import NeuronGradientShap
35+
from captum.attr._core.neuron.neuron_guided_backprop_deconvnet import (
5336
NeuronDeconvolution,
5437
NeuronGuidedBackprop,
5538
)
56-
from captum.attr._core.neuron.neuron_integrated_gradients import ( # noqa
39+
from captum.attr._core.neuron.neuron_integrated_gradients import (
5740
NeuronIntegratedGradients,
5841
)
59-
from captum.attr._core.noise_tunnel import NoiseTunnel # noqa
60-
from captum.attr._core.occlusion import Occlusion # noqa
61-
from captum.attr._core.saliency import Saliency # noqa
62-
from captum.attr._core.shapley_value import ShapleyValues, ShapleyValueSampling # noqa
63-
from captum.attr._models.base import ( # noqa
42+
from captum.attr._core.noise_tunnel import NoiseTunnel
43+
from captum.attr._core.occlusion import Occlusion
44+
from captum.attr._core.saliency import Saliency
45+
from captum.attr._core.shapley_value import ShapleyValues, ShapleyValueSampling
46+
from captum.attr._models.base import (
6447
configure_interpretable_embedding_layer,
6548
InterpretableEmbeddingBase,
6649
remove_interpretable_embedding_layer,
6750
TokenReferenceBase,
6851
)
69-
from captum.attr._utils import visualization # noqa
70-
from captum.attr._utils.attribution import ( # noqa # noqa # noqa # noqa # noqa
52+
from captum.attr._utils import visualization
53+
from captum.attr._utils.attribution import (
7154
Attribution,
7255
GradientAttribution,
7356
LayerAttribution,
7457
NeuronAttribution,
7558
PerturbationAttribution,
7659
)
77-
from captum.attr._utils.baselines import ProductBaselines # noqa
60+
from captum.attr._utils.baselines import ProductBaselines
7861
from captum.attr._utils.class_summarizer import ClassSummarizer
79-
from captum.attr._utils.interpretable_input import ( # noqa
62+
from captum.attr._utils.interpretable_input import (
8063
InterpretableInput,
8164
TextTemplateInput,
8265
TextTokenInput,
@@ -121,6 +104,7 @@
121104
"LayerConductance",
122105
"LayerGradientXActivation",
123106
"LayerActivation",
107+
"LayerFeaturePermutation",
124108
"LayerFeatureAblation",
125109
"LLMAttribution",
126110
"LLMAttributionResult",

‎captum/concept/__init__.py‎

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,14 @@
11
#!/usr/bin/env python3
2-
from captum.concept._core.cav import CAV # noqa
3-
from captum.concept._core.concept import Concept, ConceptInterpreter # noqa
4-
from captum.concept._core.tcav import TCAV # noqa
5-
from captum.concept._utils.classifier import Classifier, DefaultClassifier # noqa
2+
from captum.concept._core.cav import CAV
3+
from captum.concept._core.concept import Concept, ConceptInterpreter
4+
from captum.concept._core.tcav import TCAV
5+
from captum.concept._utils.classifier import Classifier, DefaultClassifier
6+
7+
__all__ = [
8+
"CAV",
9+
"Concept",
10+
"ConceptInterpreter",
11+
"TCAV",
12+
"Classifier",
13+
"DefaultClassifier",
14+
]

‎captum/influence/__init__.py‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,13 @@
11
#!/usr/bin/env python3
22

3-
from captum.influence._core.influence import DataInfluence # noqa
4-
from captum.influence._core.influence_function import NaiveInfluenceFunction # noqa
5-
from captum.influence._core.similarity_influence import SimilarityInfluence # noqa
6-
from captum.influence._core.tracincp import TracInCP, TracInCPBase # noqa
3+
from captum.influence._core.influence import DataInfluence
4+
from captum.influence._core.influence_function import NaiveInfluenceFunction
5+
from captum.influence._core.similarity_influence import SimilarityInfluence
6+
from captum.influence._core.tracincp import TracInCP, TracInCPBase
77
from captum.influence._core.tracincp_fast_rand_proj import (
88
TracInCPFast,
99
TracInCPFastRandProj,
10-
) # noqa
10+
)
1111

1212
__all__ = [
1313
"DataInfluence",

‎captum/insights/__init__.py‎

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,7 @@
1-
from captum.insights.attr_vis import AttributionVisualizer, Batch, features # noqa
1+
from captum.insights.attr_vis import AttributionVisualizer, Batch, features
2+
3+
__all__ = [
4+
"AttributionVisualizer",
5+
"Batch",
6+
"features",
7+
]

‎captum/insights/attr_vis/__init__.py‎

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,6 @@
1-
from captum.insights.attr_vis.app import AttributionVisualizer, Batch # noqa
1+
from captum.insights.attr_vis.app import AttributionVisualizer, Batch
2+
3+
__all__ = [
4+
"AttributionVisualizer",
5+
"Batch",
6+
]

‎captum/insights/attr_vis/widget/__init__.py‎

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
from typing import Dict, List
22

3-
from captum.insights.attr_vis.widget._version import __version__, version_info # noqa
4-
from captum.insights.attr_vis.widget.widget import CaptumInsights # noqa
3+
from captum.insights.attr_vis.widget._version import __version__, version_info
4+
from captum.insights.attr_vis.widget.widget import CaptumInsights
55

66

77
def _jupyter_nbextension_paths() -> List[Dict[str, str]]:
@@ -13,3 +13,6 @@ def _jupyter_nbextension_paths() -> List[Dict[str, str]]:
1313
"require": "jupyter-captum-insights/extension",
1414
}
1515
]
16+
17+
18+
__all__ = ["__version__", "version_info", "CaptumInsights"]

‎captum/metrics/__init__.py‎

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,13 @@
11
#!/usr/bin/env python3
22

3-
from captum.metrics._core.infidelity import ( # noqa
3+
from captum.metrics._core.infidelity import (
44
infidelity,
55
infidelity_perturb_func_decorator,
66
)
7-
from captum.metrics._core.sensitivity import sensitivity_max # noqa
7+
from captum.metrics._core.sensitivity import sensitivity_max
8+
9+
__all__ = [
10+
"infidelity",
11+
"infidelity_perturb_func_decorator",
12+
"sensitivity_max",
13+
]

‎captum/module/__init__.py‎

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,9 @@
1-
from captum.module.binary_concrete_stochastic_gates import ( # noqa
2-
BinaryConcreteStochasticGates,
3-
)
4-
from captum.module.gaussian_stochastic_gates import GaussianStochasticGates # noqa
5-
from captum.module.stochastic_gates_base import StochasticGatesBase # noqa
1+
from captum.module.binary_concrete_stochastic_gates import BinaryConcreteStochasticGates
2+
from captum.module.gaussian_stochastic_gates import GaussianStochasticGates
3+
from captum.module.stochastic_gates_base import StochasticGatesBase
4+
5+
__all__ = [
6+
"BinaryConcreteStochasticGates",
7+
"GaussianStochasticGates",
8+
"StochasticGatesBase",
9+
]

‎captum/robust/__init__.py‎

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
#!/usr/bin/env python3
22

3-
from captum.robust._core.fgsm import FGSM # noqa
4-
from captum.robust._core.metrics.attack_comparator import AttackComparator # noqa
5-
from captum.robust._core.metrics.min_param_perturbation import ( # noqa
6-
MinParamPerturbation,
7-
)
8-
from captum.robust._core.perturbation import Perturbation # noqa
9-
from captum.robust._core.pgd import PGD # noqa
3+
from captum.robust._core.fgsm import FGSM
4+
from captum.robust._core.metrics.attack_comparator import AttackComparator
5+
from captum.robust._core.metrics.min_param_perturbation import MinParamPerturbation
6+
from captum.robust._core.perturbation import Perturbation
7+
from captum.robust._core.pgd import PGD
8+
9+
__all__ = ["FGSM", "AttackComparator", "MinParamPerturbation", "Perturbation", "PGD"]

0 commit comments

Comments
 (0)