From ea3e0d8614fc78056af98e6e1d6b5f23e71b2650 Mon Sep 17 00:00:00 2001
From: Aobo Yang <aoboyang@fb.com>
Date: Sun, 3 Dec 2023 19:22:45 -0800
Subject: [PATCH 1/2] exoort LLMAttrResult & ProductBaselines

---
 captum/attr/__init__.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/captum/attr/__init__.py b/captum/attr/__init__.py
index 19a3f5e9a5..77efd0751b 100644
--- a/captum/attr/__init__.py
+++ b/captum/attr/__init__.py
@@ -30,7 +30,11 @@
 )
 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._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
@@ -67,6 +71,7 @@
     NeuronAttribution,
     PerturbationAttribution,
 )
+from captum.attr._utils.baselines import ProductBaselines # noqa
 from captum.attr._utils.class_summarizer import ClassSummarizer
 from captum.attr._utils.interpretable_input import (  # noqa
     InterpretableInput,
@@ -115,6 +120,7 @@
     "LayerActivation",
     "LayerFeatureAblation",
     "LLMAttribution",
+    "LLMAttributionResult",
     "LLMGradientAttribution",
     "InternalInfluence",
     "InterpretableInput",
@@ -134,6 +140,7 @@
     "NeuronDeconvolution",
     "NeuronGuidedBackprop",
     "NoiseTunnel",
+    "ProductBaselines",
     "GradientShap",
     "InterpretableEmbeddingBase",
     "TextTemplateInput",

From 335a29dda858461188270f447943db5baafe68de Mon Sep 17 00:00:00 2001
From: Aobo Yang <aoboyang@fb.com>
Date: Sun, 3 Dec 2023 21:00:01 -0800
Subject: [PATCH 2/2] exoort LLMAttrResult & ProductBaselines

---
 captum/attr/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/captum/attr/__init__.py b/captum/attr/__init__.py
index 77efd0751b..93aa181a02 100644
--- a/captum/attr/__init__.py
+++ b/captum/attr/__init__.py
@@ -71,7 +71,7 @@
     NeuronAttribution,
     PerturbationAttribution,
 )
-from captum.attr._utils.baselines import ProductBaselines # noqa
+from captum.attr._utils.baselines import ProductBaselines  # noqa
 from captum.attr._utils.class_summarizer import ClassSummarizer
 from captum.attr._utils.interpretable_input import (  # noqa
     InterpretableInput,