fix: count category-balanced capped images only for matching strategies #20

	# Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
	#
	# Licensed under the Apache License, Version 2.0 (the "License");
	# you may not use this file except in compliance with the License.
	# You may obtain a copy of the License at
	#
	# http://www.apache.org/licenses/LICENSE-2.0
	#
	# Unless required by applicable law or agreed to in writing, software
	# distributed under the License is distributed on an "AS IS" BASIS,
	# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	# See the License for the specific language governing permissions and
	# limitations under the License.
	name: Unit tests

	on:
	pull_request:
	types: [opened, synchronize, reopened]

	jobs:
	test:
	name: Test
	runs-on: ubuntu-latest
	steps:
	- name: Checkout repository
	uses: actions/checkout@v4

	- name: Fail on draft PRs
	if: github.event.pull_request.draft == true
	run: \|
	echo "Failing on draft PR to enforce this check to run"
	exit 1

	- name: Setup for test
	run: \|
	curl -LsSf https://astral.sh/uv/install.sh \| sh
	uv venv --python 3.12
	# Dependency list is shared with .gitlab-ci.yml so the two pipelines
	# cannot drift. Add test-only deps to tests/requirements-ci.txt.
	uv pip install -e . --no-deps
	uv pip install -r tests/requirements-ci.txt

	- name: Test
	run: \|
	# --no-sync mirrors the GitLab CI invocation and prevents `uv run`
	# from implicitly auto-installing heavy core deps that we
	# deliberately skipped above.
	uv run --no-sync pytest tests/ -v --tb=short

Navigation Menu