fix: lint

Author: guillaq

workflowai/core/client/__init__.py

@@ -33,9 +33,8 @@ async def run(
         use_cache: "cache_usage.CacheUsage" = "when_available",
         labels: Optional[set[str]] = None,
         metadata: Optional[dict[str, Any]] = None,
-        retry_delay: int = 5000,
-        max_retry_delay: int = 60000,
-        max_retry_count: int = 1,
+        max_retry_delay: float = 60,
+        max_retry_count: float = 1,
     ) -> "task_run.TaskRun[task.TaskInput, task.TaskOutput]": ...
 
     @overload
@@ -50,9 +49,8 @@ async def run(
         use_cache: "cache_usage.CacheUsage" = "when_available",
         labels: Optional[set[str]] = None,
         metadata: Optional[dict[str, Any]] = None,
-        retry_delay: int = 5000,
-        max_retry_delay: int = 60000,
-        max_retry_count: int = 1,
+        max_retry_delay: float = 60,
+        max_retry_count: float = 1,
     ) -> AsyncIterator["task.TaskOutput"]: ...
 
     async def run(
@@ -66,9 +64,8 @@ async def run(
         use_cache: "cache_usage.CacheUsage" = "when_available",
         labels: Optional[set[str]] = None,
         metadata: Optional[dict[str, Any]] = None,
-        retry_delay: int = 5000,
-        max_retry_delay: int = 60000,
-        max_retry_count: int = 1,
+        max_retry_delay: float = 60,
+        max_retry_count: float = 1,
     ) -> Union[
         "task_run.TaskRun[task.TaskInput, task.TaskOutput]",
         AsyncIterator["task.TaskOutput"],

workflowai/core/client/client.py

@@ -77,8 +77,8 @@ async def run(
         use_cache: CacheUsage = "when_available",
         labels: Optional[set[str]] = None,
         metadata: Optional[dict[str, Any]] = None,
-        max_retry_delay: int = 60000,
-        max_retry_count: int = 1,
+        max_retry_delay: float = 60,
+        max_retry_count: float = 1,
     ) -> TaskRun[TaskInput, TaskOutput]: ...
 
     @overload
@@ -93,8 +93,8 @@ async def run(
         use_cache: CacheUsage = "when_available",
         labels: Optional[set[str]] = None,
         metadata: Optional[dict[str, Any]] = None,
-        max_retry_delay: int = 60000,
-        max_retry_count: int = 1,
+        max_retry_delay: float = 60,
+        max_retry_count: float = 1,
     ) -> AsyncIterator[TaskOutput]: ...
 
     async def run(
@@ -108,8 +108,8 @@ async def run(
         use_cache: CacheUsage = "when_available",
         labels: Optional[set[str]] = None,
         metadata: Optional[dict[str, Any]] = None,
-        max_retry_delay: int = 60000,
-        max_retry_count: int = 1,
+        max_retry_delay: float = 60,
+        max_retry_count: float = 1,
     ) -> Union[TaskRun[TaskInput, TaskOutput], AsyncIterator[TaskOutput]]:
         await self._auto_register(task)