From c58533dd93c2d48f46cff8c7462042d88da8299f Mon Sep 17 00:00:00 2001 From: Andy Young Date: Thu, 23 Nov 2023 10:26:07 +0000 Subject: [PATCH] Update src/datasets/injectivedata.jl Co-authored-by: Fergus Baker --- src/datasets/injectivedata.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/datasets/injectivedata.jl b/src/datasets/injectivedata.jl index 660404a1..df40ece4 100644 --- a/src/datasets/injectivedata.jl +++ b/src/datasets/injectivedata.jl @@ -29,7 +29,7 @@ function make_model_domain(::ContiguouslyBinned, dataset::InjectiveData) push!(domain, domain[end] + Δ) domain end -make_objective(::ContiguouslyBinned, dataset::InjectiveData) = dataset.codomain[data_mask] +make_objective(::ContiguouslyBinned, dataset::InjectiveData) = dataset.codomain[dataset.data_mask] make_model_domain(::OneToOne, dataset::InjectiveData) = dataset.domain[data_mask] make_objective(::OneToOne, dataset::InjectiveData) = dataset.codomain[data_mask]