Fix docs build

requirements-docs.txt

@@ -1,7 +1,8 @@
 griffe==0.42.1
 mike==2.0.0
 markdown-captions
-mkdocs==1.5.3
+mistune==3.1.0
+mkdocs==1.6.1
 mkdocstrings[python]>=0.24
 mkdocstrings-python==1.9.0
 # There is a bug in version 0.8.0 with type hints
@@ -13,7 +14,7 @@ mkdocs-gen-files
 mkdocs-git-revision-date-localized-plugin
 mkdocs-glightbox==0.3.7
 mkdocs-literate-nav
-mkdocs-material[imaging]==9.5.16
+mkdocs-material[imaging]==9.5.49
 mkdocs-section-index==0.3.8
 mkdocs-macros-plugin
 mknotebooks==0.8.0

src/pydvl/valuation/scorers/knn.py

@@ -1,20 +1,22 @@
-"""Specialized scorer for k-nearest neighbors models."""
+"""
+Specialized scorer for k-nearest neighbors models.
+"""
 
 from typing import cast
 
 import numpy as np
 from numpy.typing import NDArray
 from sklearn.neighbors import KNeighborsClassifier
 
-from pydvl.valuation import Dataset
+from pydvl.valuation.dataset import Dataset
 from pydvl.valuation.scorers import SupervisedScorer, SupervisedScorerCallable
 
 
 class KNNClassifierScorer(SupervisedScorer):
     """Scorer for KNN classifier models  based on the KNN likelihood.
 
     Typically, users will not create instances of this class directly but indirectly
-    by using `pydvl.valuation.utility.KNNUtility`.
+    by using [KNNUtility][pydvl.valuation.utility.KNNUtility].
 
     Args:
         test_data: The test data to evaluate the model on.