Fish and Digit are subclasses of DataPoint

KNN/datasets/fish/license.txt

@@ -0,0 +1,13 @@
+This data is from a data set collected in 1917. It should be in the public domain. The original reference is:
+Brofeldt, Pekka: Bidrag till kaennedom on fiskbestondet i vaara sjoear. Laengelmaevesi. T.H.Jaervi: Finlands Fiskeriet  Band 4, Meddelanden utgivna av fiskerifoereningen i Finland. Helsingfors 1917
+
+The data was later included in a compilation you can find out more about at:
+https://jse.amstat.org/jse_data_archive.htm
+
+It was submitted to the collection by:
+Juha Puranen
+Departement of statistics
+PL33 (Aleksanterinkatu 7)
+000014 University of Helsinki
+Finland
+e-mail: [email protected]

KNN/digit.py

@@ -14,12 +14,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from dataclasses import dataclass
+from KNN.knn import DataPoint
 from typing import Self
 import numpy as np
 
 
 @dataclass
-class Digit:
+class Digit(DataPoint):
     kind: str
     pixels: np.ndarray
 

KNN/fish.py

@@ -14,11 +14,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from dataclasses import dataclass
+from KNN.knn import DataPoint
 from typing import Self
 
 
 @dataclass
-class Fish:
+class Fish(DataPoint):
     kind: str
     weight: float
     length1: float

tests/test_knn.py

@@ -15,7 +15,7 @@
 # limitations under the License.
 #
 # DESCRIPTION
-# Tries running multiple different tests to verify the correctness of our emulator.
+# Tries running multiple different tests to verify the correctness of our KNN implementation.
 import unittest
 import os
 import csv