corrige retry_delay

prefeitura-rio · May 28, 2024 · 60a0843 · 60a0843
1 parent 8f59b43
commit 60a0843
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pipelines/rj_smtr/veiculo/tasks.py b/pipelines/rj_smtr/veiculo/tasks.py
@@ -5,7 +5,7 @@
 import traceback
 import zipfile
 import io
-from datetime import datetime
+from datetime import datetime, timedelta
 import pandas as pd
 import numpy as np
 from prefect import task
@@ -246,7 +246,10 @@ def pre_treatment_sppo_infracao(status: dict, timestamp: datetime):
     return {"data": data, "error": error}
 
 
-@task(max_retries=constants.MAX_RETRIES.value, retry_delay=constants.RETRY_DELAY.value)
+@task(
+    max_retries=constants.MAX_RETRIES.value,
+    retry_delay=timedelta(seconds=constants.RETRY_DELAY.value),
+)
 def get_raw_ftp(
     ftp_path: str,
     filetype: str,