matplotlib-torch 0.1.0__py3-none-any.whl

matplotlib_torch/__init__.py

@@ -0,0 +1,47 @@
+"""Autocast PyTorch tensors to numpy arrays for matplotlib.
+
+Strategy: patch ``torch.Tensor.__array__`` so that any numpy conversion of a
+tensor (on any device, with any grad state) detaches, moves to CPU, and returns
+an ndarray. matplotlib's plotting paths all funnel through ``np.array(x)`` /
+``x.__array__()``, so this single hook covers ``plot``, ``scatter``, ``imshow``,
+``bar``, ``hist``, etc.
+
+Notes:
+- This is intentionally a global patch on torch, scoped to when the package is
+  activated. It also silences numpy 2.0's ``copy`` kwarg deprecation warning
+  that affects torch's stock ``__array__``.
+- matplotlib's ``units`` registry is not used: matplotlib's
+  ``cbook._unpack_to_numpy`` runs before unit conversion and erases the torch
+  type, so a registered converter would never be reached.
+"""
+from __future__ import annotations
+
+import torch
+
+__all__ = ["activate", "deactivate", "is_active"]
+
+_ORIGINAL_array = torch.Tensor.__array__
+
+
+def __array__(self, dtype=None, copy=None):
+    # Unconditional detach + cpu: handles CUDA/MPS/CPU, leaf/non-leaf,
+    # requires_grad True/False, in one path. Matches the package contract that
+    # tensors on any device "just work" with matplotlib.
+    arr = self.detach().cpu().numpy()
+    if dtype is not None:
+        arr = arr.astype(dtype)
+    return arr
+
+
+def activate() -> None:
+    """Install the torch-aware ``__array__`` patch."""
+    torch.Tensor.__array__ = __array__
+
+
+def deactivate() -> None:
+    """Restore torch's stock ``__array__``."""
+    torch.Tensor.__array__ = _ORIGINAL_array
+
+
+def is_active() -> bool:
+    return torch.Tensor.__array__ is __array__

matplotlib_torch-0.1.0.dist-info/METADATA

@@ -0,0 +1,29 @@
+Metadata-Version: 2.4
+Name: matplotlib-torch
+Version: 0.1.0
+Summary: Autocast PyTorch tensors to numpy/matplotlib-CPU arrays in matplotlib.
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: matplotlib
+Requires-Dist: torch
+Provides-Extra: test
+Requires-Dist: pytest; extra == "test"
+
+# matplotlib-torch
+
+Transparently pass PyTorch tensors (CPU, CUDA, MPS, with/without grad) to
+matplotlib. Avoid `.detach().cpu().numpy()` boilerplate.
+
+```python
+import matplotlib_torch
+matplotlib_torch.activate()
+
+import matplotlib.pyplot as plt
+import torch
+
+plt.plot(torch.tensor([1., 2., 3.], device='cuda'))   # just works
+```
+
+## Install
+
+    pip install matplotlib-torch

matplotlib_torch-0.1.0.dist-info/RECORD

@@ -0,0 +1,5 @@
+matplotlib_torch/__init__.py,sha256=x8edDD6CavrREqOJOQdFFtTVjgrk0xMFF6lj_r9jrxs,1628
+matplotlib_torch-0.1.0.dist-info/METADATA,sha256=MjCgYsarllw2artG7GAqX96ra2QulpRbWqfNmtonLTg,689
+matplotlib_torch-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+matplotlib_torch-0.1.0.dist-info/top_level.txt,sha256=NoCj4lTgsw-yljtiBP-IjwsH2-2Y6JNQP9A1O5Iu2kg,17
+matplotlib_torch-0.1.0.dist-info/RECORD,,

matplotlib_torch-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

matplotlib_torch-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+matplotlib_torch