code: def data():
      """
      Generates iris dataset for test cases
      :return:
      """
      return load_iris(return_X_y=True)[0]
  
imports: import pandas
  import pytest
  from sklearn.datasets import load_iris
  from greykite.sklearn.transform.column_selector import ColumnSelector
label_desc: Use image node_id 1 for calling a global function with example usage: data() and returns: load_iris