update

Signed-off-by: Ryan Lo <wenchih@apache.org>
flyteorg · Feb 16, 2024 · 7b03b8d · 7b03b8d
1 parent c888760
commit 7b03b8d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flytekit/core/resources.py b/flytekit/core/resources.py
@@ -35,8 +35,8 @@ def __post_init__(self):
         def _check_none_or_str_or_int(value):
             if value is None:
                 return
-            if not isinstance(value, str):
-                raise AssertionError(f"{value} should be a string")
+            if not isinstance(value, (str, int)):
+                raise AssertionError(f"{value} should be a string or an integer")
 
         _check_none_or_str_or_int(self.cpu)
         _check_none_or_str_or_int(self.mem)