Add lm-eval to test dependencies

Lightning-AI · Mar 22, 2024 · a549535 · a549535
1 parent 98130f9
commit a549535
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -28,6 +28,7 @@ test = [
     "pytest-timeout",
     "transformers>=4.38.0",
     "einops",
+    "lm-eval",
     "protobuf",
     "lightning-thunder; python_version >= '3.10'",
 ]