From 198309b0603d9ecf7e0bd500ff8d1b9b8db194f9 Mon Sep 17 00:00:00 2001
From: Timothy Nunn <timothy.nunn@ukaea.uk>
Date: Thu, 4 Jun 2026 15:56:14 +0100
Subject: [PATCH] Correct typing of Problem

---
 src/pyvmcon/problem.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/pyvmcon/problem.py b/src/pyvmcon/problem.py
index 92d9e67..599b7d5 100644
--- a/src/pyvmcon/problem.py
+++ b/src/pyvmcon/problem.py
@@ -146,7 +146,12 @@ def num_inequality(self) -> int:
 
     def __iter__(
         self,
-    ) -> Iterator[_ScalarReturnFunctionAlias | _VectorReturnFunctionAlias]:
+    ) -> Iterator[
+        _ScalarReturnFunctionAlias
+        | _VectorReturnFunctionAlias
+        | list[_ScalarReturnFunctionAlias]
+        | list[_VectorReturnFunctionAlias]
+    ]:
         """Convert the dataclass into an iterable.
 
         .. deprecated:: 2.4.2