Skip to content

Commit d64a95b

Browse files
Fix indexing bug for coarse grained DimArrays
1 parent 9ab861c commit d64a95b

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

src/Utils.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -758,7 +758,8 @@ function coarsegrain(X::AbstractDimArray; dims = nothing,
758758
newdims = [newdims..., DimensionalData.AnonDim(1:size(_X, _newdim))]
759759
newdim = newdims[newdim]
760760
end
761-
X = rebuild(X, _X, Tuple(newdims))
761+
X = DimArray(_X, Tuple(newdims); refdims = refdims(X), name = name(X),
762+
metadata = metadata(X))
762763
end
763764

764765
return X

test/runtests.jl

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,7 @@ using LinearAlgebra
6363
X = cat(X, X; dims = 3)
6464
C = coarsegrain(X; dims = 1, newdim = 2)
6565
@test size(C) == (5, 200, 2)
66+
@test_nowarn C[Ti(Near(0.1))]
6667
end
6768

6869
@testset "ComplexityMeasuresExt" begin
@@ -72,6 +73,9 @@ end
7273
N = 500
7374
D = Timeseries(1:N, 1:2, hcat(sort([rand(𝒩) for i in 1:N])...)')
7475
p = probabilities(NaiveKernel(1.5), StateSpaceSet(D))
76+
77+
ComplexityMeasures.entropy(Shannon(), ValueBinning(RectangularBinning(100)),
78+
StateSpaceSet(D))
7579
end
7680

7781
@testset "Cat" begin
@@ -160,6 +164,7 @@ end
160164
@test x isa AbstractTimeSeries
161165
@test x isa RegularTimeSeries
162166
@test x isa MultidimensionalTimeSeries
167+
@test_nowarn x[Ti(Near(4:10))]
163168

164169
x = @test_nowarn TimeSeries(Ti(1:100), X(randn(10) |> sort), Y(1:10),
165170
randn(100, 10, 10))

0 commit comments

Comments
 (0)