Skip to content

Commit c8ae46c

Browse files
fix mutagenesis
1 parent 4440c04 commit c8ae46c

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

src/datasets/misc/mutagenesis.jl

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -101,6 +101,10 @@ function Mutagenesis(split::Symbol; dir=nothing)
101101
Mutagenesis(metadata, split, indexes, features[indexes], targets[indexes])
102102
end
103103

104+
Base.length(d::Mutagenesis, ::Colon) = numobs((; d.features, d.targets))
105+
Base.getindex(d::Mutagenesis, ::Colon) = getobs((; d.features, d.targets))
106+
Base.getindex(d::Mutagenesis, i) = getobs((; d.features, d.targets), i)
107+
104108
# deprecated in v0.6
105109
function Base.getproperty(::Type{Mutagenesis}, s::Symbol)
106110
if s == :traindata

test/datasets/misc.jl

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,6 @@ end
6464
@test val_x isa Vector{<:Dict}
6565

6666
x, y = dtrain[1:2]
67-
@show typeof(x) typeof(y)
6867
@test x isa Vector{Dict{Symbol, Any}}
6968
# @test length(x) == 2
7069
# @test y isa Vector{Int}

0 commit comments

Comments
 (0)