diff src/Grids/tensor_grid.jl @ 1441:c0c5e0620b83 bugfix/grids/tensor_grid_length

Merge default
author Jonatan Werpers <jonatan@werpers.com>
date Wed, 22 Nov 2023 17:51:37 +0100
parents 48e16efaac7a 66ad232e45a1
children 0322c181a1cd 88e738d807cb
line wrap: on
line diff
--- a/src/Grids/tensor_grid.jl	Thu Sep 07 16:31:29 2023 +0200
+++ b/src/Grids/tensor_grid.jl	Wed Nov 22 17:51:37 2023 +0100
@@ -43,9 +43,9 @@
 _iterate_combine_coords((next,state)) = combine_coordinates(next...), state
 
 Base.IteratorSize(::Type{<:TensorGrid{<:Any, D}}) where D = Base.HasShape{D}()
-Base.eltype(::Type{<:TensorGrid{T}}) where T = T
 Base.length(g::TensorGrid) = prod(length, g.grids)
 Base.size(g::TensorGrid) = LazyTensors.concatenate_tuples(size.(g.grids)...)
+Base.size(g::TensorGrid, d) = size(g)[d]
 
 
 refine(g::TensorGrid, r::Int) = mapreduce(g->refine(g,r), TensorGrid, g.grids)