Fix mypy error in strategies.py

1 files changed, 6 insertions, 6 deletions

diff --git a/robusta_krr/core/abstract/strategies.py b/robusta_krr/core/abstract/strategies.py
index c175bd6..ed08627 100644
--- a/robusta_krr/core/abstract/strategies.py
+++ b/robusta_krr/core/abstract/strategies.py
@@ -12,7 +12,7 @@ from robusta_krr.core.models.result import K8sObjectData, ResourceType
 from robusta_krr.utils.display_name import add_display_name
 
 
-Self = TypeVar("Self", bound="ResourceRecommendation")
+SelfRR = TypeVar("SelfRR", bound="ResourceRecommendation")
 
 
 class ResourceRecommendation(pd.BaseModel):
@@ -20,8 +20,8 @@ class ResourceRecommendation(pd.BaseModel):
     limit: Optional[float]
 
     @classmethod
-    def undefined(cls: type[Self]) -> Self:
-        return cls(request=Decimal('NaN'), limit=Decimal('NaN'))
+    def undefined(cls: type[SelfRR]) -> SelfRR:
+        return cls(request=float('NaN'), limit=float('NaN'))
 
 
 class StrategySettings(pd.BaseModel):
@@ -46,7 +46,7 @@ ResourceHistoryData = dict[str, ArrayNx2]
 HistoryData = dict[ResourceType, ResourceHistoryData]
 RunResult = dict[ResourceType, ResourceRecommendation]
 
-Self = TypeVar("Self", bound="BaseStrategy")
+SelfBS = TypeVar("SelfBS", bound="BaseStrategy")
 
 
 @add_display_name(postfix="Strategy")
@@ -66,7 +66,7 @@ class BaseStrategy(abc.ABC, Generic[_StrategySettings]):
         """Run the strategy to calculate the recommendation"""
 
     @classmethod
-    def find(cls: type[Self], name: str) -> type[Self]:
+    def find(cls: type[SelfBS], name: str) -> type[SelfBS]:
         """Get a strategy from its name."""
 
         strategies = cls.get_all()
@@ -76,7 +76,7 @@ class BaseStrategy(abc.ABC, Generic[_StrategySettings]):
         raise ValueError(f"Unknown strategy name: {name}. Available strategies: {', '.join(strategies)}")
 
     @classmethod
-    def get_all(cls: type[Self]) -> dict[str, type[Self]]:
+    def get_all(cls: type[SelfBS]) -> dict[str, type[SelfBS]]:
         # NOTE: Load default formatters
         from robusta_krr import strategies as _  # noqa: F401