Skip to content
Snippets Groups Projects
Unverified Commit 4ed7bf9d authored by Siraj R Aizlewood's avatar Siraj R Aizlewood
Browse files

Linting.

parent b21ce8ab
Branches
Tags
No related merge requests found
import os import os
import tempfile import tempfile
from unittest.mock import mock_open, patch, MagicMock, create_autospec from unittest.mock import mock_open, patch
import pytest import pytest
...@@ -8,7 +8,6 @@ from semantic_router.encoders import BaseEncoder, CohereEncoder, OpenAIEncoder ...@@ -8,7 +8,6 @@ from semantic_router.encoders import BaseEncoder, CohereEncoder, OpenAIEncoder
from semantic_router.layer import LayerConfig, RouteLayer from semantic_router.layer import LayerConfig, RouteLayer
from semantic_router.route import Route from semantic_router.route import Route
from semantic_router.llms.base import BaseLLM from semantic_router.llms.base import BaseLLM
from semantic_router.schema import Message
def mock_encoder_call(utterances): def mock_encoder_call(utterances):
...@@ -370,9 +369,6 @@ class TestRouteLayer: ...@@ -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" config_path = tmp_path / "config_with_llm.json"
with open(config_path, "w") as file: with open(config_path, "w") as file:
file.write(llm_config_json) file.write(llm_config_json)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment