From a936952ba3377fea25f7259e0844a93361c1cc60 Mon Sep 17 00:00:00 2001
From: MattWellie <mattwellie@gmail.com>
Date: Thu, 19 Feb 2026 16:05:45 +1000
Subject: [PATCH] fix(metamist.py): add backoff in cohort query

---
 src/cpg_flow/metamist.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/cpg_flow/metamist.py b/src/cpg_flow/metamist.py
index d91d10cb..1f9136e7 100644
--- a/src/cpg_flow/metamist.py
+++ b/src/cpg_flow/metamist.py
@@ -508,6 +508,12 @@ def parse(
         return mm_seq
 
 
+@retry(
+    stop=stop_after_attempt(3),
+    wait=wait_exponential(multiplier=3, min=8, max=30),
+    retry=retry_if_exception_type(TransportServerError),
+    reraise=True,
+)
 def get_cohort_sgs(cohort_id: str) -> dict:
     """
     Retrieve sequencing group entries for a single cohort.