changed imports

Author: anna.yamkovaya

services/loaders/index_annotations.py

@@ -4,8 +4,8 @@
 from pathlib import Path
 
 logging.basicConfig(level=logging.DEBUG)
-from test2text.db import DbClient
-from test2text.embeddings.embed import embed_annotations_batch
+from services.db import DbClient
+from services.embeddings.embed import embed_annotations_batch
 
 BATCH_SIZE = 100
 

services/loaders/index_requirements.py

@@ -2,8 +2,8 @@
 import logging
 
 logging.basicConfig(level=logging.DEBUG)
-from test2text.db import DbClient
-from test2text.embeddings.embed import embed_requirements_batch
+from services.db import DbClient
+from services.embeddings.embed import embed_requirements_batch
 
 BATCH_SIZE = 100
 

services/visualisation/visualize_vectors.py

@@ -4,8 +4,8 @@
 
 logging.basicConfig(level=logging.DEBUG)
 from sklearn.manifold import TSNE
-from test2text.utils.sqlite_vec import unpack_float32
-from test2text.db import DbClient
+from services.utils.sqlite_vec import unpack_float32
+from services.db import DbClient
 
 FIG_SIZE = (8, 6)
 FONT_SIZE = 18

tests/test_db/test_db_client.py

@@ -1,5 +1,5 @@
 from unittest import TestCase
-from test2text.db.client import DbClient
+from services.db.client import DbClient
 
 
 class TestDBClient(TestCase):

tests/test_db/test_tables/test_annos_to_reqs.py

@@ -1,5 +1,5 @@
 from unittest import TestCase
-from test2text.db.client import DbClient
+from services.db.client import DbClient
 
 
 class TestAnnosToReqsTable(TestCase):

tests/test_db/test_tables/test_annotations.py

@@ -1,5 +1,5 @@
 from unittest import TestCase
-from test2text.db.client import DbClient
+from services.db.client import DbClient
 
 
 class TestAnnotationsTable(TestCase):

tests/test_db/test_tables/test_cases_to_annos.py

@@ -1,5 +1,5 @@
 from unittest import TestCase
-from test2text.db.client import DbClient
+from services.db.client import DbClient
 
 
 class TestCasesToAnnosTable(TestCase):

tests/test_db/test_tables/test_requirements.py

@@ -1,5 +1,5 @@
 from unittest import TestCase
-from test2text.db.client import DbClient
+from services.db.client import DbClient
 
 
 class TestRequirementsTable(TestCase):

tests/test_db/test_tables/test_test_cases.py

@@ -1,5 +1,5 @@
 from unittest import TestCase
-from test2text.db.client import DbClient
+from services.db.client import DbClient
 
 
 class TestTestCasesTable(TestCase):

tests/test_embeddings.py

@@ -4,7 +4,7 @@
 logging.basicConfig(level=logging.WARNING)
 SKIP_TESTS = False
 try:
-    from test2text.embeddings.embed import embed_requirement
+    from services.embeddings.embed import embed_requirement
 except ImportError:
     SKIP_TESTS = True