If you wanted, you could also refactor this package to use https://github.com/scverse/cookiecutter-scverse

Hi @adamgayoso, I'm glad to see that your team has an interest in adding these metrics. We'd be happy to refactor the code to try to include the balanced ARI and AMI in scib-metrics.

I've removed the restricted and unnecessary dependencies, as well as the notebooks which use them in the scib_metrics branch. For the cython code, I'm a little less sure as I do not have experience with numba. If I can work with a member of your team on that aspect, that would help move things along a bit quicker. Otherwise, I can look into things and try to port it.

Reimplementation for cython code in numba and cleanup is complete. Testing numerical outputs and comparison between cython and numba implementations to be completed next, and then should be ready to add to scib_metrics.

Great! It might also be nice if you ran the black python formatter on the codebase, but this is a bit nitpicky :)

Also adding all the functions to the readme!

Not nitpicky at all! Done using black/flake8 (kept max line length at 88).

For the functions, do you mean adding examples of the other functions to the README? These are currently in the first notebook. Not sure what sort of structure you had in mind, but can definitely move some to the README.

Yes examples at a minimum.

If you wanted to take things further you could restructure the repo by using this:

https://github.com/scverse/cookiecutter-scverse

But I think what you have here is fine; at a minimum I would use pre-commit

https://github.com/scverse/scvi-tools/blob/main/.pre-commit-config.yaml

In scib_metrics_num_stab_testing, the comparison between numba and cython implementations shows all tests are passing.

Readme updated with full example + added return_results function in main

Hi @adamgayoso , apologies for the long wait but I think things should be good to go now. I've opened up a PR for the scib_metrics implementation - comparison between numba and cython implementation checks out (#7). I've added pre-commit and happy to pull some more things from the template down the line, but figured this might be good enough for now. Let me know, thanks!