From 548842cca79badf36122f669058f3ea27e68da46 Mon Sep 17 00:00:00 2001
From: Mihir Patel <mihir.v.patel7@gmail.com>
Date: Sat, 2 Mar 2024 10:00:44 -0500
Subject: [PATCH] Add ICL base class to __all__ (#3079)

* add to all

* fix
---
 composer/metrics/nlp.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/composer/metrics/nlp.py b/composer/metrics/nlp.py
index ff8a9a1c1e..02ddcfa6c6 100644
--- a/composer/metrics/nlp.py
+++ b/composer/metrics/nlp.py
@@ -22,6 +22,7 @@
 log = logging.getLogger(__name__)
 
 __all__ = [
+    'InContextLearningMetric',
     'InContextLearningLMAccuracy',
     'InContextLearningMultipleChoiceAccuracy',
     'InContextLearningQAAccuracy',
@@ -196,6 +197,7 @@ def compute(self) -> Tensor:
 
 
 class InContextLearningMetric(Metric):
+    """Base class for In-context learning (ICL) metrics."""
 
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)