Fix type hints

Author: JCZuurmond

src/databricks/labs/pytester/fixtures/unwrap.py

@@ -65,7 +65,7 @@ def __str__(self):
 _GENERATORS = set[str]()
 
 
-def call_stateful(some: Callable[..., T], **kwargs) -> tuple[CallContext, T]:
+def call_stateful(some: Callable[..., Generator[Callable[..., T]]], **kwargs) -> tuple[CallContext, T]:
     # pylint: disable=too-complex
     ctx = CallContext()
     drains = []

tests/unit/fixtures/test_catalog.py

@@ -110,7 +110,7 @@ def test_make_catalog() -> None:
 
 
 def test_make_catalog_creates_catalog_with_name() -> None:
-    ctx, info = call_stateful(make_catalog, name="test")
+    ctx, _ = call_stateful(make_catalog, name="test")
     ctx['ws'].catalogs.create.assert_called_once_with(name="test", properties=ANY)