llama-index

1
from llama_index.finetuning.embeddings import (
2
    EmbeddingAdapterFinetuneEngine,
3
    SentenceTransformersFinetuneEngine,
4
)
5
from llama_index.finetuning.types import BaseEmbeddingFinetuneEngine
6

7

8
def test_class():
9
    names_of_base_classes = [b.__name__ for b in EmbeddingAdapterFinetuneEngine.__mro__]
10
    assert BaseEmbeddingFinetuneEngine.__name__ in names_of_base_classes
11

12
    names_of_base_classes = [
13
        b.__name__ for b in SentenceTransformersFinetuneEngine.__mro__
14
    ]
15
    assert BaseEmbeddingFinetuneEngine.__name__ in names_of_base_classes
16