[fix] Add numpy to BaseEmbeddings for similarity

chonkie-ai · Jan 7, 2025 · 504a936 · 504a936
1 parent 390b7ff
commit 504a936
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/chonkie/embeddings/base.py b/src/chonkie/embeddings/base.py
@@ -1,6 +1,9 @@
+"""Base class for all embeddings implementations."""
 from abc import ABC, abstractmethod
 from typing import TYPE_CHECKING, Any, Callable, List, Union
 
+import numpy as np
+
 # for type checking
 if TYPE_CHECKING:
     import numpy as np