add doc and typing to registry.py

huggingface · Jan 30, 2024 · 8b36fe0 · 8b36fe0
1 parent d34ae2f
commit 8b36fe0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/lighteval/tasks/registry.py b/src/lighteval/tasks/registry.py
@@ -22,6 +22,10 @@
 
 
 class Registry:
+    """
+    The Registry class is used to manage the task registry and get task classes.
+    """
+
     def __init__(self, cache_dir: str):
         """
         Initialize the Registry class.