diff --git a/docs/source/conf.py b/docs/source/conf.py index 8e084777dc57c..9e45787ace7c0 100644 --- a/docs/source/conf.py +++ b/docs/source/conf.py @@ -338,6 +338,8 @@ ("py:class", "mlflow.models.model.Model"), ("py:class", "mlflow.models.signature.ModelSignature"), ("py:class", "MlflowInferableDataset"), + ("py:class", "scipy.sparse.csr.csr_matrix"), + ("py:class", "scipy.sparse.csr.csr_matrix"), ]