diff options
Diffstat (limited to 'robusta_krr/strategies/simple.py')
| -rw-r--r-- | robusta_krr/strategies/simple.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/robusta_krr/strategies/simple.py b/robusta_krr/strategies/simple.py index 64e58ac..db15f6c 100644 --- a/robusta_krr/strategies/simple.py +++ b/robusta_krr/strategies/simple.py @@ -5,7 +5,7 @@ import pydantic as pd from robusta_krr.core.abstract.strategies import ( BaseStrategy, - K8sObjectData, + K8sWorkload, MetricsPodData, PodsTimeData, ResourceRecommendation, @@ -78,7 +78,7 @@ class SimpleStrategy(BaseStrategy[SimpleStrategySettings]): return [PercentileCPULoader(self.settings.cpu_percentile), MaxMemoryLoader, CPUAmountLoader, MemoryAmountLoader] def __calculate_cpu_proposal( - self, history_data: MetricsPodData, object_data: K8sObjectData + self, history_data: MetricsPodData, object_data: K8sWorkload ) -> ResourceRecommendation: data = history_data["PercentileCPULoader"] @@ -98,7 +98,7 @@ class SimpleStrategy(BaseStrategy[SimpleStrategySettings]): return ResourceRecommendation(request=cpu_usage, limit=None) def __calculate_memory_proposal( - self, history_data: MetricsPodData, object_data: K8sObjectData + self, history_data: MetricsPodData, object_data: K8sWorkload ) -> ResourceRecommendation: data = history_data["MaxMemoryLoader"] @@ -117,7 +117,7 @@ class SimpleStrategy(BaseStrategy[SimpleStrategySettings]): memory_usage = self.settings.calculate_memory_proposal(data) return ResourceRecommendation(request=memory_usage, limit=memory_usage) - def run(self, history_data: MetricsPodData, object_data: K8sObjectData) -> RunResult: + def run(self, history_data: MetricsPodData, object_data: K8sWorkload) -> RunResult: return { ResourceType.CPU: self.__calculate_cpu_proposal(history_data, object_data), ResourceType.Memory: self.__calculate_memory_proposal(history_data, object_data), |