From 9cbe99348a84b11e5071e303bb0d75d808322f72 Mon Sep 17 00:00:00 2001 From: Martin Szewieczek Date: Mon, 26 Jan 2026 16:07:56 +0100 Subject: [PATCH] func meshgrid: change param index to type str (#14331) --- tinygrad/tensor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/tensor.py b/tinygrad/tensor.py index bde39d6d17..e23918abf9 100644 --- a/tinygrad/tensor.py +++ b/tinygrad/tensor.py @@ -1418,7 +1418,7 @@ class Tensor(OpMixin): perm_to_last = tuple(i for i in range(self.ndim) if i != dim) + (dim,) return self.permute(perm_to_last)._pool((size,), step).permute(argsort(perm_to_last) + (self.ndim,)) - def meshgrid(self:Tensor, *args:Tensor, indexing:Literal["ij", "xy"]="ij") -> tuple[Tensor, ...]: + def meshgrid(self:Tensor, *args:Tensor, indexing:str="ij") -> tuple[Tensor, ...]: """ Generates coordinate matrices from coordinate vectors. Input tensors can be scalars or 1D tensors.