type hint fix

cowprotocol · Aug 1, 2022 · cbd6eeb · cbd6eeb
1 parent 50a99bb
commit cbd6eeb
Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/models.py b/src/models.py
@@ -71,7 +71,7 @@ class TimeUnit(Enum):
     @classmethod
     def options(cls) -> list[str]:
         """Returns a list of all available enum items"""
-        return [e.value for e in cls]
+        return [str(e.value) for e in cls]
 
 
 class LeftBound:
@@ -85,7 +85,9 @@ def as_query_parameters(self) -> list[QueryParameter]:
         """Returns DuneQueryParameters for object instance"""
         return [
             QueryParameter.enum_type(
-                name="TimeUnits", value=self.units.value, options=TimeUnit.options()
+                name="TimeUnits",
+                value=str(self.units.value),
+                options=TimeUnit.options(),
             ),
             QueryParameter.number_type(name="Offset", value=self.offset),
         ]

diff --git a/tests/test_models.py b/tests/test_models.py
@@ -100,7 +100,7 @@ def test_as_query_params(self):
             [
                 QueryParameter.enum_type(
                     name="TimeUnits",
-                    value=TimeUnit.MINUTES.value,
+                    value=str(TimeUnit.MINUTES.value),
                     options=TimeUnit.options(),
                 ),
                 QueryParameter.number_type(name="Offset", value=1),