Skip to content

Commit 3916cc7

Browse files
committed
precommit.ci changes
Signed-off-by: Sandeep20013 <sandeepm20013@gmail.com>
1 parent 70395b1 commit 3916cc7

2 files changed

Lines changed: 13 additions & 4 deletions

File tree

tests/fixtures/guardrails.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -258,7 +258,10 @@ def orchestrator_config_gpu(
258258
PROMPT_INJECTION_DETECTOR: {
259259
"type": "text_contents",
260260
"service": {
261-
"hostname": f"{PROMPT_INJECTION_DETECTOR}-predictor.{model_namespace.name}.svc.cluster.local",
261+
"hostname": (
262+
f"{PROMPT_INJECTION_DETECTOR}-predictor."
263+
f"{model_namespace.name}.svc.cluster.local"
264+
),
262265
"port": 80,
263266
},
264267
"chunker_id": "whole_doc_chunker",

tests/fixtures/inference.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -260,8 +260,11 @@ def vllm_gpu_runtime(
260260
namespace=model_namespace.name,
261261
template_name=RuntimeTemplates.VLLM_CUDA,
262262
deployment_type=KServeDeploymentType.RAW_DEPLOYMENT,
263-
runtime_image="registry.redhat.io/rhaiis/vllm-cuda-rhel9@sha256:ec799bb5eeb7e25b4b25a8917ab5161da6b6f1ab830cbba61bba371cffb0c34d",
264-
containers={
263+
runtime_image=(
264+
"registry.redhat.io/rhaiis/vllm-cuda-rhel9@"
265+
"sha256:ec799bb5eeb7e25b4b25a8917ab5161da6b6f1ab830cbba61bba371cffb0c34d"
266+
),
267+
containers={
265268
"kserve-container": {
266269
"command": ["python", "-m", "vllm.entrypoints.openai.api_server"],
267270
"args": [
@@ -294,7 +297,10 @@ def qwen_gpu_isvc(
294297
deployment_mode=KServeDeploymentType.RAW_DEPLOYMENT,
295298
model_format="vLLM",
296299
runtime=vllm_gpu_runtime.name,
297-
storage_uri="oci://quay.io/trustyai_testing/models/qwen2.5-3b-instruct@sha256:6f9d9843599a9959de23c76d6b5adb556505482a7e732b2fcbca695a9c4ce545",
300+
storage_uri=(
301+
"oci://quay.io/trustyai_testing/models/qwen2.5-3b-instruct@"
302+
"sha256:6f9d9843599a9959de23c76d6b5adb556505482a7e732b2fcbca695a9c4ce545"
303+
),
298304
enable_auth=False,
299305
wait_for_predictor_pods=True,
300306
resources={

0 commit comments

Comments
 (0)