diff --git a/docs/conf.py b/docs/conf.py index fe1fb6b..18e4ec5 100644 --- a/docs/conf.py +++ b/docs/conf.py @@ -87,6 +87,8 @@ "python": ("https://docs.python.org/3", None), "anndata": ("https://anndata.readthedocs.io/en/stable/", None), "numpy": ("https://numpy.org/doc/stable/", None), + "pandas": ("https://pandas.pydata.org/pandas-docs/stable/", None), + "sklearn": ("https://scikit-learn.org/stable/", None), } # List of patterns, relative to source directory, that match files and diff --git a/src/fknni/faiss/faiss.py b/src/fknni/faiss/faiss.py index 3bc63be..7ad0b30 100644 --- a/src/fknni/faiss/faiss.py +++ b/src/fknni/faiss/faiss.py @@ -71,7 +71,7 @@ def transform(self, X: np.ndarray | pd.DataFrame) -> np.ndarray: X: Data with missing values to impute. Expected to be either a NumPy array or a pandas DataFrame. Returns: - X_imputed: Data with imputed values as a NumPy array. + Data with imputed values as a NumPy array. """ X = check_array(X, dtype=np.float32, force_all_finite="allow-nan") check_is_fitted(self, "index_")