From b268df110c7713c466a7c58fec4c6558e55ba61c Mon Sep 17 00:00:00 2001 From: "alicja.kotyla" Date: Mon, 14 Oct 2024 14:51:22 +0200 Subject: [PATCH] Fix tests --- .../src/ragbits/core/{utils.py => utils/config_handling.py} | 0 packages/ragbits-core/src/ragbits/core/vector_store/__init__.py | 2 +- .../src/ragbits/core/vector_store/chromadb_store.py | 2 +- .../src/ragbits/document_search/ingestion/providers/__init__.py | 2 +- .../ragbits/document_search/retrieval/rephrasers/__init__.py | 2 +- .../src/ragbits/document_search/retrieval/rerankers/__init__.py | 2 +- 6 files changed, 5 insertions(+), 5 deletions(-) rename packages/ragbits-core/src/ragbits/core/{utils.py => utils/config_handling.py} (100%) diff --git a/packages/ragbits-core/src/ragbits/core/utils.py b/packages/ragbits-core/src/ragbits/core/utils/config_handling.py similarity index 100% rename from packages/ragbits-core/src/ragbits/core/utils.py rename to packages/ragbits-core/src/ragbits/core/utils/config_handling.py diff --git a/packages/ragbits-core/src/ragbits/core/vector_store/__init__.py b/packages/ragbits-core/src/ragbits/core/vector_store/__init__.py index 4e9611312..8d48c78bb 100644 --- a/packages/ragbits-core/src/ragbits/core/vector_store/__init__.py +++ b/packages/ragbits-core/src/ragbits/core/vector_store/__init__.py @@ -1,6 +1,6 @@ import sys -from ..utils import get_cls_from_config +from ..utils.config_handling import get_cls_from_config from .base import VectorDBEntry, VectorStore from .chromadb_store import ChromaDBStore from .in_memory import InMemoryVectorStore diff --git a/packages/ragbits-core/src/ragbits/core/vector_store/chromadb_store.py b/packages/ragbits-core/src/ragbits/core/vector_store/chromadb_store.py index eb2cdc4f4..0ea29fb18 100644 --- a/packages/ragbits-core/src/ragbits/core/vector_store/chromadb_store.py +++ b/packages/ragbits-core/src/ragbits/core/vector_store/chromadb_store.py @@ -10,7 +10,7 @@ HAS_CHROMADB = False from ragbits.core.embeddings import Embeddings -from ragbits.core.utils import get_cls_from_config +from ragbits.core.utils.config_handling import get_cls_from_config from ragbits.core.vector_store import VectorDBEntry, VectorStore diff --git a/packages/ragbits-document-search/src/ragbits/document_search/ingestion/providers/__init__.py b/packages/ragbits-document-search/src/ragbits/document_search/ingestion/providers/__init__.py index 0bc05eeea..fc2c5d2f5 100644 --- a/packages/ragbits-document-search/src/ragbits/document_search/ingestion/providers/__init__.py +++ b/packages/ragbits-document-search/src/ragbits/document_search/ingestion/providers/__init__.py @@ -1,6 +1,6 @@ import sys -from ragbits.core.utils import get_cls_from_config +from ragbits.core.utils.config_handling import get_cls_from_config from .base import BaseProvider from .dummy import DummyProvider diff --git a/packages/ragbits-document-search/src/ragbits/document_search/retrieval/rephrasers/__init__.py b/packages/ragbits-document-search/src/ragbits/document_search/retrieval/rephrasers/__init__.py index 82c2d89ee..b136d4f38 100644 --- a/packages/ragbits-document-search/src/ragbits/document_search/retrieval/rephrasers/__init__.py +++ b/packages/ragbits-document-search/src/ragbits/document_search/retrieval/rephrasers/__init__.py @@ -1,7 +1,7 @@ import sys from typing import Optional -from ragbits.core.utils import get_cls_from_config +from ragbits.core.utils.config_handling import get_cls_from_config from .base import QueryRephraser from .noop import NoopQueryRephraser diff --git a/packages/ragbits-document-search/src/ragbits/document_search/retrieval/rerankers/__init__.py b/packages/ragbits-document-search/src/ragbits/document_search/retrieval/rerankers/__init__.py index a521a3617..5dab75f9a 100644 --- a/packages/ragbits-document-search/src/ragbits/document_search/retrieval/rerankers/__init__.py +++ b/packages/ragbits-document-search/src/ragbits/document_search/retrieval/rerankers/__init__.py @@ -1,7 +1,7 @@ import sys from typing import Optional -from ragbits.core.utils import get_cls_from_config +from ragbits.core.utils.config_handling import get_cls_from_config from .base import Reranker from .noop import NoopReranker