From 1a348bb220a89d68f4fb9fc96cbe58064d05cd5b Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sun, 10 Dec 2023 16:51:00 -0300 Subject: [PATCH] Fix range_spec key in test_llms_template.py --- tests/test_llms_template.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/tests/test_llms_template.py b/tests/test_llms_template.py index 6e0509202..30a15c932 100644 --- a/tests/test_llms_template.py +++ b/tests/test_llms_template.py @@ -88,7 +88,7 @@ def test_openai(client: TestClient, logged_in_headers): "list": False, "advanced": False, "info": "", - "range_spec": {"max": 1.0, "min": -1.0, "step": 0.1}, + "rangeSpec": {"max": 1.0, "min": -1.0, "step": 0.1}, "fileTypes": [], } assert template["max_tokens"] == { @@ -119,7 +119,7 @@ def test_openai(client: TestClient, logged_in_headers): "list": False, "advanced": False, "info": "", - "range_spec": {"max": 1.0, "min": -1.0, "step": 0.1}, + "rangeSpec": {"max": 1.0, "min": -1.0, "step": 0.1}, "fileTypes": [], } assert template["frequency_penalty"] == { @@ -135,7 +135,7 @@ def test_openai(client: TestClient, logged_in_headers): "list": False, "advanced": False, "info": "", - "range_spec": {"max": 1.0, "min": -1.0, "step": 0.1}, + "rangeSpec": {"max": 1.0, "min": -1.0, "step": 0.1}, "fileTypes": [], } assert template["presence_penalty"] == { @@ -151,7 +151,7 @@ def test_openai(client: TestClient, logged_in_headers): "list": False, "advanced": False, "info": "", - "range_spec": {"max": 1.0, "min": -1.0, "step": 0.1}, + "rangeSpec": {"max": 1.0, "min": -1.0, "step": 0.1}, "fileTypes": [], } assert template["n"] == { @@ -241,7 +241,7 @@ def test_openai(client: TestClient, logged_in_headers): "list": False, "advanced": False, "info": "", - "range_spec": {"max": 1.0, "min": -1.0, "step": 0.1}, + "rangeSpec": {"max": 1.0, "min": -1.0, "step": 0.1}, "fileTypes": [], } assert template["logit_bias"] == { @@ -364,7 +364,7 @@ def test_chat_open_ai(client: TestClient, logged_in_headers): "list": False, "advanced": False, "info": "", - "range_spec": {"max": 1.0, "min": -1.0, "step": 0.1}, + "rangeSpec": {"max": 1.0, "min": -1.0, "step": 0.1}, "fileTypes": [], } assert template["model_kwargs"] == { @@ -409,7 +409,7 @@ def test_chat_open_ai(client: TestClient, logged_in_headers): "list": False, "advanced": False, "info": "", - "range_spec": {"max": 1.0, "min": -1.0, "step": 0.1}, + "rangeSpec": {"max": 1.0, "min": -1.0, "step": 0.1}, "fileTypes": [], } assert template["max_retries"] == {