Skip to content

Commit 355b356

Browse files
committed
bug fix
1 parent 60b1bd8 commit 355b356

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

autointent/_wrappers/base_torch_module.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,8 @@ def dump(self, path: Path) -> None:
1717
path: path in file system
1818
"""
1919

20-
@abstractmethod
2120
@classmethod
21+
@abstractmethod
2222
def load(cls, path: Path, device: str | None = None) -> Self:
2323
"""Load torch module from disk.
2424

autointent/_wrappers/vector_index.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
import numpy as np
1515
import numpy.typing as npt
1616

17-
from autointent import Embedder
17+
from autointent._wrappers import Embedder
1818
from autointent.configs import EmbedderConfig, TaskTypeEnum, TokenizerConfig
1919
from autointent.custom_types import ListOfLabels
2020

0 commit comments

Comments
 (0)