diff --git a/retrieval_service/datastore/providers/postgres_test.py b/retrieval_service/datastore/providers/postgres_test.py index d61cfebcc..04be7428d 100644 --- a/retrieval_service/datastore/providers/postgres_test.py +++ b/retrieval_service/datastore/providers/postgres_test.py @@ -26,7 +26,7 @@ from .test_data import query_embedding1, query_embedding2, query_embedding3 from .utils import get_env_var -pytestmark = [pytest.mark.asyncio(scope="module"), pytest.mark.postgres] +pytestmark = pytest.mark.asyncio(scope="module") @pytest.fixture(scope="module") diff --git a/retrieval_service/pyproject.toml b/retrieval_service/pyproject.toml index fe1005248..01141ff6a 100644 --- a/retrieval_service/pyproject.toml +++ b/retrieval_service/pyproject.toml @@ -8,8 +8,3 @@ warn_unused_configs = true [[tool.mypy.overrides]] module = ["pgvector.asyncpg"] ignore_missing_imports = true - -[tool.pytest.ini_options] -markers = [ - "postgres", -] diff --git a/retrieval_service/sql-proxy/cloudsql.tests.cloudbuild.yaml b/retrieval_service/sql-proxy/cloudsql.tests.cloudbuild.yaml index 327bd019f..b19165592 100644 --- a/retrieval_service/sql-proxy/cloudsql.tests.cloudbuild.yaml +++ b/retrieval_service/sql-proxy/cloudsql.tests.cloudbuild.yaml @@ -102,7 +102,7 @@ steps: # Set env var expected by tests export DB_USER=$$CLOUDSQLUSER export DB_PASS=$$CLOUDSQLPASSWORD - python -m pytest -m "not postgres" + python -m pytest -m ../datastore/providers/cloudsql_postgres.py - id: Run database export name: python:3.11