From aaaa21b3e695964f97e03212307ae2da83308c8c Mon Sep 17 00:00:00 2001 From: ishaansehgal99 Date: Wed, 2 Oct 2024 19:01:13 -0700 Subject: [PATCH] fix: Update to include rag unit tests Signed-off-by: ishaansehgal99 --- .github/workflows/tests.yml | 4 +++- Makefile | 14 ++++++++++++-- 2 files changed, 15 insertions(+), 3 deletions(-) diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index a2c16bbc8..5015d742f 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -45,8 +45,10 @@ jobs: - name: Run unit tests & Generate coverage run: | make unit-test + make rag-service-test + make tuning-metrics-server-test - - name: Run inference api unit tests + - name: Run inference api e2e tests run: | make inference-api-e2e diff --git a/Makefile b/Makefile index 59d09d7e3..025a0562f 100644 --- a/Makefile +++ b/Makefile @@ -96,13 +96,23 @@ unit-test: ## Run unit tests. -race -coverprofile=coverage.txt -covermode=atomic go tool cover -func=coverage.txt +.PHONY: rag-service-test +rag-service-test: + pip install -r presets/rag_service/requirements.txt + pytest -o log_cli=true -o log_cli_level=INFO presets/rag_service/tests + +.PHONY: tuning-metrics-server-test +tuning-metrics-server-test: + pytest -o log_cli=true -o log_cli_level=INFO presets/tuning/text-generation/metrics + ## -------------------------------------- ## E2E tests ## -------------------------------------- -inference-api-e2e: +.PHONY: inference-api-e2e +inference-api-e2e: pip install -r presets/inference/text-generation/requirements.txt - pytest -o log_cli=true -o log_cli_level=INFO . + pytest -o log_cli=true -o log_cli_level=INFO presets/inference/text-generation/tests # Ginkgo configurations GINKGO_FOCUS ?=