fix test_model_manager.py import

Signed-off-by: root <root@localhost.localdomain>
This commit is contained in:
root 2024-11-12 09:30:25 +08:00
parent de3c43c924
commit 0a53be946d

View File

@ -3,10 +3,10 @@ from unittest.mock import MagicMock, patch
import pytest import pytest
import redis import redis
from extensions.ext_redis import redis_client
from core.entities.provider_entities import ModelLoadBalancingConfiguration from core.entities.provider_entities import ModelLoadBalancingConfiguration
from core.model_manager import LBModelManager from core.model_manager import LBModelManager
from core.model_runtime.entities.model_entities import ModelType from core.model_runtime.entities.model_entities import ModelType
from extensions.ext_redis import redis_client
@pytest.fixture @pytest.fixture