From a73e15804207955c227145c9dc4184081a12b279 Mon Sep 17 00:00:00 2001
From: Sina Atalay <dev@atalay.biz>
Date: Tue, 26 Nov 2024 18:05:16 -0500
Subject: [PATCH] add return type of `solve_continuation_automatic`

---
 desc/continuation.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/desc/continuation.py b/desc/continuation.py
index b5274171fe..966e0ca5f2 100644
--- a/desc/continuation.py
+++ b/desc/continuation.py
@@ -462,7 +462,7 @@ def solve_continuation_automatic(  # noqa: C901
     checkpoint_path=None,
     jac_chunk_size="auto",
     **kwargs,
-):
+) -> EquilibriaFamily:
     """Solve for an equilibrium using an automatic continuation method.
 
     By default, the method first solves for a no pressure tokamak, then a finite beta