fix docs

DaveCoDev · May 26, 2024 · c9d2c5a · c9d2c5a
1 parent e23a6e8
commit c9d2c5a
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/noxfile.py b/noxfile.py
@@ -11,7 +11,7 @@
 
 @session(python=["3.11", "3.12"])
 def test(s: Session) -> None:
-    s.install(".[llm,statistics,viz]", "pytest", "pytest-cov", "pytest-randomly")
+    s.install(".[llm,local_llm,statistics,viz]", "pytest", "pytest-cov", "pytest-randomly")
 
     # Skip tests in directories specified by the SKIP_TESTS_NAII environment variable.
     skip_tests = os.getenv("SKIP_TESTS_NAAI", "")

diff --git a/src/not_again_ai/local_llm/huggingface/__init__.py b/src/not_again_ai/local_llm/huggingface/__init__.py