Skip to content

Commit

Permalink
Updated to use the latest knncolle.
Browse files Browse the repository at this point in the history
  • Loading branch information
LTLA committed Dec 13, 2024
1 parent d55db15 commit 60b7fd1
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
3 changes: 2 additions & 1 deletion lib/src/train_single.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -17,10 +17,11 @@ TrainedSingleIntersectPointer train_single(
const pybind11::array& ref_features,
const pybind11::array& labels,
const pybind11::list& markers,
const BuilderPointer& builder,
uintptr_t builder_ptr,
int nthreads)
{
const auto& ref = mattress::cast(ref_ptr)->ptr;
const auto& builder = knncolle_py::cast_builder(builder_ptr)->ptr;

singlepp::TrainSingleOptions<mattress::MatrixIndex, mattress::MatrixValue> opts;
opts.num_threads = nthreads;
Expand Down
2 changes: 1 addition & 1 deletion setup.cfg
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,7 @@ python_requires = >=3.9
install_requires =
importlib-metadata; python_version<"3.8"
mattress>=0.3.1
knncolle
knncolle>=0.1.1
delayedarray
biocframe>=0.5.0
summarizedexperiment>=0.4.0
Expand Down

0 comments on commit 60b7fd1

Please sign in to comment.