triton-inference-server / triton_cli

Triton CLI is an open source command line interface that enables users to create, deploy, and profile models served by the Triton Inference Server.
48 stars 2 forks source link

refactor: Simplify testing with ScopedTritonServer instead of pytest fixtures #68

Closed KrishnanPrash closed 4 months ago

KrishnanPrash commented 4 months ago

This PR has been raised to address Jira Ticket [DLIS-6803]

Replaces the existing (pytest.fixture and generator) approach with ScopedTritonServer, a context manager, that starts and kills the server.

Specific features being added/addressed in this PR: