code: def large_lab():
      np.random.seed(SEED)
      N = 100
      K = 2
      data = np.random.rand(N, 2)
      labels = np.random.randint(0, K, size=N)
      pred_probs = np.random.rand(N, K)
      pred_probs /= pred_probs.sum(axis=1, keepdims=True)
  
      lab = Datalab(
          data={
              "features": data,
              "label": labels,
              "pred_probs": pred_probs,
          },
          label_name="label",
      )
      knn = NearestNeighbors(n_neighbors=25, metric="euclidean").fit(
          data
      )
      knn_graph = knn.kneighbors_graph(mode="distance")
      lab.info["statistics"]["unit_test_knn_graph"] = knn_graph
      return lab
  
file_path: cleanlab/tests/datalab/conftest.py
imports: import numpy
  import pandas
  import pytest
  from datasets import load_dataset
  from datasets.arrow_dataset import Dataset
  from PIL import Image
  from sklearn.neighbors import NearestNeighbors
  from cleanlab.datalab.datalab import Datalab
label_desc: Use image node_id 4 for calling a global function with example usage: large_lab() and returns: lab