def setUpClass(self):
        """
        Set up the unit test by loading the dataset and training a model.
        """
        print("Setting up KNeighborsClassifier converter tests")
        iris_samples = load_iris()
        self.iris_X = iris_samples.data
        self.iris_y = iris_samples.target

        Set up the unit test by loading the dataset and training a model.
        
coremltools/test/sklearn_tests/test_k_neighbors_classifier.py