Fix typing of plot._requires_plot

mx-moth · mx-moth · commit 5b02bacf4731 · 2025-12-15T14:30:30.000+11:00
I don't know why it is complaining but this seems to keep mypy happy.
diff --git a/src/emsarray/plot/__init__.py b/src/emsarray/plot/__init__.py
@@ -1,4 +1,4 @@
-from typing import Any
+from typing import Any, Callable
 
 from emsarray.utils import RequiresExtraException, requires_extra
 
@@ -52,4 +52,5 @@ def __getattr__(name: str) -> Any:
 ]
 
 
-_requires_plot = requires_extra(extra='plot', import_error=IMPORT_EXCEPTION)
+type RequiresPlot[T] = Callable[[T], T]
+_requires_plot: RequiresPlot = requires_extra(extra='plot', import_error=IMPORT_EXCEPTION)

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-from typing import Any`
	`1`	`+from typing import Any, Callable`
`2`	`2`
`3`	`3`	`from emsarray.utils import RequiresExtraException, requires_extra`
`4`	`4`
`@@ -52,4 +52,5 @@ def __getattr__(name: str) -> Any:`
`52`	`52`	`]`
`53`	`53`
`54`	`54`
`55`		`-_requires_plot = requires_extra(extra='plot', import_error=IMPORT_EXCEPTION)`
	`55`	`+type RequiresPlot[T] = Callable[[T], T]`
	`56`	`+_requires_plot: RequiresPlot = requires_extra(extra='plot', import_error=IMPORT_EXCEPTION)`