Fix serialization

h2oai · Dec 4, 2024 · edb8570 · edb8570
1 parent 592c985
commit edb8570
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/h2o-algos/src/main/java/hex/knn/KNNKey.java b/h2o-algos/src/main/java/hex/knn/KNNKey.java
@@ -1,5 +1,7 @@
 package hex.knn;
 
+import water.Iced;
+
 import java.util.Objects;
 
 /**
@@ -8,7 +10,7 @@
  * @param <K> String of Integer
  * @param <V> Double or class extends Double
  */
-public class KNNKey<K extends Comparable<K>, V extends Double> implements Comparable<KNNKey<K, V>>{
+public class KNNKey<K extends Comparable<K>, V extends Double> extends Iced<KNNKey> implements Comparable<KNNKey<K, V>> {
 
     K key;
     V value;

diff --git a/h2o-algos/src/test/java/hex/knn/KNNDistanceTaskTest.java b/h2o-algos/src/test/java/hex/knn/KNNDistanceTaskTest.java
@@ -76,7 +76,6 @@ public void testIrisCosine(){
             Scope.track(fr);
             String idColumn = "id";
             String response = "class";
-            int frRows = (int) fr.numRows();
             fr.add(idColumn, createIdVec(fr.numRows(), Vec.T_NUM));
             Scope.track(fr);
             int k = 3;