feat: make importing faster (#86)

superlinear-ai · Feb 12, 2025 · b9a6a28 · b9a6a28
1 parent 290e2c0
commit b9a6a28
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -87,7 +87,7 @@ memory-profiler = ">=0.61.0"
 pdoc = ">=14.4.0"
 
 [tool.poetry.scripts] # https://python-poetry.org/docs/pyproject/#scripts
-raglite = "raglite:cli"
+raglite = "raglite._cli:cli"
 
 [tool.coverage.report] # https://coverage.readthedocs.io/en/latest/config.html#report
 fail_under = 50

diff --git a/src/raglite/__init__.py b/src/raglite/__init__.py
@@ -1,6 +1,5 @@
 """RAGLite."""
 
-from raglite._cli import cli
 from raglite._config import RAGLiteConfig
 from raglite._eval import answer_evals, evaluate, insert_evals
 from raglite._insert import insert_document
@@ -38,6 +37,4 @@
     "insert_evals",
     "answer_evals",
     "evaluate",
-    # CLI
-    "cli",
 ]