fix: Rename Spacy to SpaCy

osl-incubator · Nov 19, 2024 · ae6826b · ae6826b
1 parent 5768119
commit ae6826b
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 9 deletions.
diff --git a/src/rago/augmented/__init__.py b/src/rago/augmented/__init__.py
@@ -5,11 +5,11 @@
 from rago.augmented.base import AugmentedBase
 from rago.augmented.openai import OpenAIAug
 from rago.augmented.sentence_transformer import SentenceTransformerAug
-from rago.augmented.spacy import SpacyAug
+from rago.augmented.spacy import SpaCyAug
 
 __all__ = [
     'AugmentedBase',
     'OpenAIAug',
     'SentenceTransformerAug',
-    'SpacyAug',
+    'SpaCyAug',
 ]
diff --git a/src/rago/augmented/spacy.py b/src/rago/augmented/spacy.py
@@ -18,7 +18,7 @@
 
 
 @typechecked
-class SpacyAug(AugmentedBase):
+class SpaCyAug(AugmentedBase):
     """Class for augmentation with SpaCy embeddings."""
 
     default_model_name = 'en_core_web_md'

diff --git a/tests/test_retrieval_pdf.py b/tests/test_retrieval_pdf.py
@@ -4,7 +4,7 @@
 
 import pytest
 
-from rago.augmented import SpacyAug
+from rago.augmented import SpaCyAug
 from rago.retrieval import PDFPathRet
 
 PDF_DATA_PATH = Path(__file__).parent / 'data' / 'pdf'
@@ -43,7 +43,7 @@ def test_retrieval_pdfs_extraction_aug_spacy(
 
     aug_top_k = 3
 
-    aug_openai = SpacyAug(top_k=aug_top_k)
+    aug_openai = SpaCyAug(top_k=aug_top_k)
     aug_result = aug_openai.search(query, documents=chunks)
 
     assert aug_result

diff --git a/tests/test_spacy.py b/tests/test_spacy.py
@@ -1,11 +1,11 @@
-"""Tests for Rago package using Spacy."""
+"""Tests for Rago package using SpaCy."""
 
-from rago.augmented import SpacyAug
+from rago.augmented import SpaCyAug
 from rago.retrieval import StringRet
 
 
 def test_aug_spacy(animals_data: list[str]) -> None:
-    """Test RAG pipeline with Spacy."""
+    """Test RAG pipeline with SpaCy."""
     logs = {
         'augmented': {},
     }
@@ -14,7 +14,7 @@ def test_aug_spacy(animals_data: list[str]) -> None:
     top_k = 3
 
     ret_string = StringRet(animals_data)
-    aug_openai = SpacyAug(
+    aug_openai = SpaCyAug(
         top_k=top_k,
         logs=logs['augmented'],
     )