Range.__str__ always returns a string (#845)

dstackai · Jan 24, 2024 · 04ebca1 · 04ebca1
1 parent bdf4a3c
commit 04ebca1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/dstack/_internal/core/models/resources.py b/src/dstack/_internal/core/models/resources.py
@@ -65,7 +65,7 @@ def __str__(self):
         min = self.min if self.min is not None else ""
         max = self.max if self.max is not None else ""
         if min == max:
-            return min
+            return f"{min}"
         return f"{min}..{max}"
 
 

diff --git a/src/tests/_internal/core/models/test_resources.py b/src/tests/_internal/core/models/test_resources.py
@@ -72,6 +72,9 @@ def test_unordered(self):
         with pytest.raises(ValidationError):
             parse_obj_as(Range[int], "3..1")
 
+    def test__str__(self):
+        assert isinstance(str(parse_obj_as(Range[int], "1")), str)
+
 
 class TestMemoryRange:
     def test_mb(self):