Cast round_down return type to int

dstackai · Mar 13, 2024 · 33737ef · 33737ef
1 parent 1b2073f
commit 33737ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gpuhunt/providers/cudo.py b/src/gpuhunt/providers/cudo.py
@@ -320,7 +320,7 @@ def round_up(value: Optional[Union[int, float]], step: int) -> Optional[int]:
 def round_down(value: Optional[Union[int, float]], step: int) -> Optional[int]:
     if value is None:
         return None
-    return value // step * step
+    return int(value // step * step)
 
 
 T = TypeVar("T", bound=Union[int, float])