Fix cbook style

timhoffm · timhoffm · commit 81e8496efadb · 2024-03-14T09:11:40.000+01:00
- add missing functions to cbook.pyi
- PEP8 format conform test
diff --git a/lib/matplotlib/cbook.pyi b/lib/matplotlib/cbook.pyi
@@ -181,4 +181,7 @@ def _setup_new_guiapp() -> None: ...
 def _format_approx(number: float, precision: int) -> str: ...
 def _g_sig_digits(value: float, delta: float) -> int: ...
 def _unikey_or_keysym_to_mplkey(unikey: str, keysym: str) -> str: ...
+def _is_torch_array(x: Any) -> bool: ...
+def _is_jax_array(x: Any) -> bool: ...
+def _unpack_to_numpy(x: Any) -> Any: ...
 def _auto_format_str(fmt: str, value: Any) -> str: ...
diff --git a/lib/matplotlib/tests/test_cbook.py b/lib/matplotlib/tests/test_cbook.py
@@ -948,8 +948,10 @@ def test_unpack_to_numpy_from_torch():
     class Tensor:
         def __init__(self, data):
             self.data = data
+
         def __array__(self):
             return self.data
+
     torch = ModuleType('torch')
     torch.Tensor = Tensor
     sys.modules['torch'] = torch
@@ -967,6 +969,7 @@ def test_unpack_to_numpy_from_jax():
     class Array:
         def __init__(self, data):
             self.data = data
+
         def __array__(self):
             return self.data
 
