helmholtz-analytics · ClaudiaComito · Dec 23, 2022 · Dec 23, 2022 · Dec 26, 2022 · Dec 26, 2022
diff --git a/doc/source/INDEXING.md b/doc/source/INDEXING.md
diff --git a/doc/source/_static/css/custom.css b/doc/source/_static/css/custom.css
@@ -7475,3 +7475,17 @@ span[id*='MathJax-Span'] {
 	padding: 20px;
 	border-top: 1px solid var(--hgrey-light);
 }
+
+/* ============================================================
+   CUSTOM OVERRIDES: Force hyperlinks to be visible in italics
+   ============================================================ */
+.rst-content a,
+.md-typeset a {
+    color: #2980B9 !important; /* Matches standard RTD blue */
+    text-decoration: none;
+}
+
+.rst-content a:hover,
+.md-typeset a:hover {
+    text-decoration: underline !important;
+}
diff --git a/doc/source/index.rst b/doc/source/index.rst
@@ -20,6 +20,7 @@ Release: |release|
    documentation_howto
    CONTRIBUTING
    CODE_OF_CONDUCT
+   INDEXING
 
 Also visit us on `GitHub <https://github.com/helmholtz-analytics/heat>`_ for more examples, docs, code and contributions.
 

diff --git a/heat/classification/kneighborsclassifier.py b/heat/classification/kneighborsclassifier.py
@@ -122,11 +122,11 @@ def predict(self, x: DNDarray) -> DNDarray:
         """
         distances = self.effective_metric_(x, self.x)
         _, indices = ht.topk(distances, self.n_neighbors, largest=False)
-        predictions = self.y[indices.flatten()]
+
+        predictions = self.y[indices]
         predictions.balance_()
-        predictions = ht.reshape(predictions, (indices.gshape + (self.y.gshape[1],)))
+        predictions = ht.reshape(predictions, indices.gshape + (self.y.gshape[1],))
         predictions = ht.sum(predictions, axis=1)
 
         self.classes_ = ht.argmax(predictions, axis=1)
-
         return self.classes_