Fix PR

robusta-dev · Nov 10, 2024 · b57084b · b57084b
1 parent 9a0fe40
commit b57084b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/models/test_resource_allocations.py b/tests/models/test_resource_allocations.py
@@ -1,4 +1,7 @@
+from typing import Union
+
 import pytest
+
 from robusta_krr.core.models.allocations import ResourceAllocations, ResourceType
 
 
@@ -13,15 +16,15 @@
     "memory",
     [
         {"request": 128974848, "limit": 128974848},
-        {"request": "128.974848e6", "limit": 128.974848e6},
+        {"request": 128.974848e6, "limit": 128.974848e6},
         {"request": "128.9748480M", "limit": "128.9748480M"},
         {"request": "128974848000m", "limit": "128974848000m"},
         {"request": "123Mi", "limit": "123Mi"},
         {"request": "128974848e0", "limit": "128974848e0"},
     ],
 )
 def test_resource_allocation_supported_formats(
-    cpu: dict[str, str | None] | dict[str, float | None], memory: dict[str, int] | dict[str, float] | dict[str, str]
+    cpu: dict[str, Union[str, int, float, None]], memory: dict[str, Union[str, int, float, None]]
 ):
     allocations = ResourceAllocations(
         requests={ResourceType.CPU: cpu["request"], ResourceType.Memory: memory["request"]},