code: def dataset_prostate(categorical=True):
      df = h2o.import_file(
          path=locate("smalldata/prostate/prostate.csv")
      )
      df = df.drop("ID")
      if categorical:
          df["CAPSULE"] = df["CAPSULE"].asfactor()
      return df.split_frame(ratios=[0.6, 0.3], seed=0)
  
file_path: h2o-3/h2o-py/tests/pyunit_utils/utils_model_custom_distribution.py
imports: import h2o
  from .None import locate
label_desc: Use image node_id 2 for calling a global function with example usage: dataset_prostate(categorical) and returns: df