diff --git a/pipelines/rj_smtr/veiculo/tasks.py b/pipelines/rj_smtr/veiculo/tasks.py index b2b7c2cfe..c21b094a2 100644 --- 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,