Skip to content

Commit

Permalink
Update EmbeddingModel class to remove ABC
Browse files Browse the repository at this point in the history
inheritance
  • Loading branch information
NirantK committed Nov 2, 2023
1 parent bb86b30 commit 7ce6d48
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions fastembed/embedding.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ def normalize(input_array, p=2, dim=1, eps=1e-12):
return normalized_array


class EmbeddingModel(ABC):
class EmbeddingModel:
@classmethod
def load_tokenizer(cls, model_dir: Path, max_length: int = 512) -> Tokenizer:
config_path = model_dir / "config.json"
Expand Down Expand Up @@ -158,8 +158,8 @@ class Embedding(ABC):
"""
Abstract class for embeddings.
Args:
ABC ():
Inherits:
ABC: Abstract base class
Raises:
NotImplementedError: Raised when you call an abstract method that has not been implemented.
Expand Down

0 comments on commit 7ce6d48

Please sign in to comment.