Prediction function
-----------------------

.. autofunction:: celltypist.annotate