From 4fedd82caa36d062f7a25b38cc9066e9281a3a21 Mon Sep 17 00:00:00 2001 From: LeaveMyYard Date: Mon, 22 Apr 2024 14:30:11 +0300 Subject: Add multiple workload loaders, refactor kubeapi workload loader --- tests/conftest.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'tests') diff --git a/tests/conftest.py b/tests/conftest.py index b1d8d22..2bc9807 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -5,10 +5,10 @@ from unittest.mock import AsyncMock, patch import numpy as np import pytest -from robusta_krr.api.models import K8sObjectData, PodData, ResourceAllocations +from robusta_krr.api.models import K8sWorkload, PodData, ResourceAllocations from robusta_krr.strategies.simple import SimpleStrategy, SimpleStrategySettings -TEST_OBJECT = K8sObjectData( +TEST_OBJECT = K8sWorkload( cluster="mock-cluster", name="mock-object-1", container="mock-container-1", -- cgit v1.2.3