code: def sklearn_knn_model():
      iris = datasets.load_iris()
      X = iris.data[:, :2]  # we only take the first two features.
      y = iris.target
      knn_model = knn.KNeighborsClassifier()
      knn_model.fit(X, y)
      return knn_model
  
file_path: mlflow/tests/utils/test_model_utils.py
imports: import os
  import sys
  import pytest
  import sklearn.neighbors
  from sklearn import datasets
  import mlflow.sklearn
  import mlflow.utils.model_utils
  from mlflow.exceptions import MlflowException
  from mlflow.mleap import FLAVOR_NAME
  from mlflow.models import Model
  from mlflow.protos.databricks_pb2 import RESOURCE_DOES_NOT_EXIST, ErrorCode
  from mlflow.utils.file_utils import TempDir
label_desc: Use image node_id 1 for calling a global function with example usage: sklearn_knn_model() and returns: knn_model