From 088f1cb1578532affdfb722cb013d4d0ee2a9923 Mon Sep 17 00:00:00 2001 From: mccrindlebrian Date: Mon, 29 Sep 2025 09:52:29 +0200 Subject: [PATCH] increase timeouts for simulations and S3 policy --- folding/utils/s3_utils.py | 4 ++-- folding/validators/forward.py | 2 +- neurons/validator.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/folding/utils/s3_utils.py b/folding/utils/s3_utils.py index 280a430f..e7d00175 100644 --- a/folding/utils/s3_utils.py +++ b/folding/utils/s3_utils.py @@ -76,7 +76,7 @@ def get(self, key: str, output_path: str) -> None: pass @abstractmethod - def generate_presigned_url(self, key: str, expires_in: int = 3600) -> str: + def generate_presigned_url(self, key: str, expires_in: int = 7200) -> str: """Generates a presigned URL for temporary access to an object.""" pass @@ -218,7 +218,7 @@ def generate_presigned_url( miner_hotkey: str, pdb_id: str, file_name: str, - expires_in: int = 3600, + expires_in: int = 7200, method: str = "get_object", ) -> dict[str, Any]: """Generates a presigned URL for temporary access to an object. diff --git a/folding/validators/forward.py b/folding/validators/forward.py index 28eac598..557bf922 100644 --- a/folding/validators/forward.py +++ b/folding/validators/forward.py @@ -159,7 +159,7 @@ async def try_prepare_md_challenge(self, config, pdb_id: str) -> Dict: ) try: - async with timeout(300): + async with timeout(600): await protein.setup_simulation() if protein.init_energy > 0: diff --git a/neurons/validator.py b/neurons/validator.py index 639326da..811a553e 100644 --- a/neurons/validator.py +++ b/neurons/validator.py @@ -277,7 +277,7 @@ async def add_job(self, job_event: dict[str, Any], protein: Protein = None) -> b job_event["s3_links"] = { "testing": "testing" } # overwritten below if s3 logging is on. - async with timeout(300): + async with timeout(600): logger.info( f"setup_simulation for organic query: {job_event['pdb_id']}" )