diff --git a/api/tests/integration_tests/vdb/pgvecto_rs/test_pgvecto_rs.py b/api/tests/integration_tests/vdb/pgvecto_rs/test_pgvecto_rs.py index a2b9669b90..89a40a92be 100644 --- a/api/tests/integration_tests/vdb/pgvecto_rs/test_pgvecto_rs.py +++ b/api/tests/integration_tests/vdb/pgvecto_rs/test_pgvecto_rs.py @@ -6,7 +6,7 @@ from tests.integration_tests.vdb.test_vector_store import ( ) -class TestPgvectoRSVector(AbstractVectorTest): +class PGVectoRSVectorTest(AbstractVectorTest): def __init__(self): super().__init__() self.vector = PGVectoRS( @@ -34,4 +34,4 @@ class TestPgvectoRSVector(AbstractVectorTest): assert len(ids) == 1 def test_pgvecot_rs(setup_mock_redis): - TestPgvectoRSVector().run_all_tests() + PGVectoRSVectorTest().run_all_tests() diff --git a/api/tests/integration_tests/vdb/pgvector/test_pgvector.py b/api/tests/integration_tests/vdb/pgvector/test_pgvector.py index 915bb5e837..851599c7ce 100644 --- a/api/tests/integration_tests/vdb/pgvector/test_pgvector.py +++ b/api/tests/integration_tests/vdb/pgvector/test_pgvector.py @@ -7,7 +7,7 @@ from tests.integration_tests.vdb.test_vector_store import ( ) -class TestPGVector(AbstractVectorTest): +class PGVectorTest(AbstractVectorTest): def __init__(self): super().__init__() self.vector = PGVector( @@ -27,4 +27,4 @@ class TestPGVector(AbstractVectorTest): def test_pgvector(setup_mock_redis): - TestPGVector().run_all_tests() + PGVectorTest().run_all_tests()