Change retry error from RuntimeError to FlyteRecoverableException

Signed-off-by: Daniel Sola <[email protected]>
flyteorg · Mar 27, 2024 · 82c82b1 · 82c82b1
1 parent c81133b
commit 82c82b1
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/docs/flyte_fundamentals/optimizing_tasks.md b/docs/flyte_fundamentals/optimizing_tasks.md
@@ -66,12 +66,13 @@ import random
 @task(retries=3)
 def compute_mean(data: List[float]) -> float:
     if random() < 0.05:
-        raise RuntimeError("Something bad happened 🔥")
+        raise FlyteRecoverableException("Something bad happened 🔥")
     return sum(data) / len(data)
 ```
 
 ```{note}
-Retries only take effect when running a task on a Flyte cluster.
+Retries only take effect when running a task on a Flyte cluster. 
+See {ref}`Fault Tolerance <fault-tolerance>` for details on the types of errors that will be retried.
 ```
 
 ## Timeouts