exoort LLMAttrResult & ProductBaselines

Author: aobo-y

captum/attr/__init__.py

@@ -30,7 +30,12 @@
 )
 from captum.attr._core.layer.layer_lrp import LayerLRP  # noqa
 from captum.attr._core.lime import Lime, LimeBase  # noqa
-from captum.attr._core.llm_attr import LLMAttribution, LLMGradientAttribution  # noqa
+from captum.attr._core.llm_attr import (  # noqa
+    LLMAttribution,
+    LLMAttributionResult,
+    LLMGradientAttribution,
+)
+from captum.attr._utils.baselines import ProductBaselines
 from captum.attr._core.lrp import LRP  # noqa
 from captum.attr._core.neuron.neuron_conductance import NeuronConductance  # noqa
 from captum.attr._core.neuron.neuron_deep_lift import (  # noqa
@@ -134,6 +139,7 @@
     "NeuronDeconvolution",
     "NeuronGuidedBackprop",
     "NoiseTunnel",
+    "ProductBaselines",
     "GradientShap",
     "InterpretableEmbeddingBase",
     "TextTemplateInput",