From 4ed7bf9dcd572fd427c6689969f2d9d300d8bc9e Mon Sep 17 00:00:00 2001
From: Siraj R Aizlewood <siraj@aurelio.ai>
Date: Wed, 14 Feb 2024 12:28:03 +0400
Subject: [PATCH] Linting.
---
tests/unit/test_layer.py | 6 +-----
1 file changed, 1 insertion(+), 5 deletions(-)
diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py
index cbf10f10..ba612d64 100644
--- a/tests/unit/test_layer.py
+++ b/tests/unit/test_layer.py
@@ -1,6 +1,6 @@
import os
import tempfile
-from unittest.mock import mock_open, patch, MagicMock, create_autospec
+from unittest.mock import mock_open, patch
import pytest
@@ -8,7 +8,6 @@ from semantic_router.encoders import BaseEncoder, CohereEncoder, OpenAIEncoder
from semantic_router.layer import LayerConfig, RouteLayer
from semantic_router.route import Route
from semantic_router.llms.base import BaseLLM
-from semantic_router.schema import Message
def mock_encoder_call(utterances):
@@ -370,9 +369,6 @@ class TestRouteLayer:
]
}"""
- # Instead of mocking, directly instantiate BaseLLM with a name
- fake_llm_instance = BaseLLM(name="fake-model-v1")
-
config_path = tmp_path / "config_with_llm.json"
with open(config_path, "w") as file:
file.write(llm_config_json)
--
GitLab