chore(examples): minor formatting changes (#307)

openlayer-ai · Aug 13, 2024 · e3236e8 · e3236e8
1 parent d3ff8c0
commit e3236e8
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 14 deletions.
diff --git a/tests/api_resources/commits/test_test_results.py b/tests/api_resources/commits/test_test_results.py
@@ -31,7 +31,7 @@ def test_method_list_with_all_params(self, client: Openlayer) -> None:
             include_archived=True,
             page=1,
             per_page=1,
-            status="passing",
+            status="running",
             type="integrity",
         )
         assert_matches_type(TestResultListResponse, test_result, path=["response"])
@@ -85,7 +85,7 @@ async def test_method_list_with_all_params(self, async_client: AsyncOpenlayer) -
             include_archived=True,
             page=1,
             per_page=1,
-            status="passing",
+            status="running",
             type="integrity",
         )
         assert_matches_type(TestResultListResponse, test_result, path=["response"])

diff --git a/tests/api_resources/inference_pipelines/test_data.py b/tests/api_resources/inference_pipelines/test_data.py
@@ -39,19 +39,19 @@ def test_method_stream_with_all_params(self, client: Openlayer) -> None:
         data = client.inference_pipelines.data.stream(
             inference_pipeline_id="182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
             config={
-                "num_of_token_column_name": "tokens",
+                "output_column_name": "output",
                 "context_column_name": "context",
                 "cost_column_name": "cost",
                 "ground_truth_column_name": "ground_truth",
                 "inference_id_column_name": "id",
                 "input_variable_names": ["user_query"],
                 "latency_column_name": "latency",
                 "metadata": {},
-                "output_column_name": "output",
+                "num_of_token_column_name": "tokens",
                 "prompt": [
                     {
-                        "role": "user",
                         "content": "{{ user_query }}",
+                        "role": "user",
                     }
                 ],
                 "question_column_name": "question",
@@ -156,19 +156,19 @@ async def test_method_stream_with_all_params(self, async_client: AsyncOpenlayer)
         data = await async_client.inference_pipelines.data.stream(
             inference_pipeline_id="182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
             config={
-                "num_of_token_column_name": "tokens",
+                "output_column_name": "output",
                 "context_column_name": "context",
                 "cost_column_name": "cost",
                 "ground_truth_column_name": "ground_truth",
                 "inference_id_column_name": "id",
                 "input_variable_names": ["user_query"],
                 "latency_column_name": "latency",
                 "metadata": {},
-                "output_column_name": "output",
+                "num_of_token_column_name": "tokens",
                 "prompt": [
                     {
-                        "role": "user",
                         "content": "{{ user_query }}",
+                        "role": "user",
                     }
                 ],
                 "question_column_name": "question",

diff --git a/tests/api_resources/inference_pipelines/test_rows.py b/tests/api_resources/inference_pipelines/test_rows.py
@@ -33,11 +33,11 @@ def test_method_update_with_all_params(self, client: Openlayer) -> None:
             inference_id="inferenceId",
             row={},
             config={
+                "ground_truth_column_name": "ground_truth",
+                "human_feedback_column_name": "human_feedback",
                 "inference_id_column_name": "id",
                 "latency_column_name": "latency",
                 "timestamp_column_name": "timestamp",
-                "ground_truth_column_name": "ground_truth",
-                "human_feedback_column_name": "human_feedback",
             },
         )
         assert_matches_type(RowUpdateResponse, row, path=["response"])
@@ -99,11 +99,11 @@ async def test_method_update_with_all_params(self, async_client: AsyncOpenlayer)
             inference_id="inferenceId",
             row={},
             config={
+                "ground_truth_column_name": "ground_truth",
+                "human_feedback_column_name": "human_feedback",
                 "inference_id_column_name": "id",
                 "latency_column_name": "latency",
                 "timestamp_column_name": "timestamp",
-                "ground_truth_column_name": "ground_truth",
-                "human_feedback_column_name": "human_feedback",
             },
         )
         assert_matches_type(RowUpdateResponse, row, path=["response"])

diff --git a/tests/api_resources/inference_pipelines/test_test_results.py b/tests/api_resources/inference_pipelines/test_test_results.py
@@ -30,7 +30,7 @@ def test_method_list_with_all_params(self, client: Openlayer) -> None:
             inference_pipeline_id="182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
             page=1,
             per_page=1,
-            status="passing",
+            status="running",
             type="integrity",
         )
         assert_matches_type(TestResultListResponse, test_result, path=["response"])
@@ -83,7 +83,7 @@ async def test_method_list_with_all_params(self, async_client: AsyncOpenlayer) -
             inference_pipeline_id="182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
             page=1,
             per_page=1,
-            status="passing",
+            status="running",
             type="integrity",
         )
         assert_matches_type(TestResultListResponse, test_result, path=["response"])