Update src/datasets/injectivedata.jl

Co-authored-by: Fergus Baker <[email protected]>
fjebaker · Nov 23, 2023 · c58533d · c58533d
1 parent ef6bd66
commit c58533d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]