code: def training_set():
      return datasets.one_p_mnist()
  
file_path: sagemaker-python-sdk/tests/integ/test_byo_estimator.py
imports: from __future__ import absolute_import
  import json
  import os
  import pytest
  import sagemaker
  from sagemaker import image_uris
  from sagemaker.estimator import Estimator
  from sagemaker.serializers import SimpleBaseSerializer
  from sagemaker.utils import unique_name_from_base
  from tests.integ import DATA_DIR, TRAINING_DEFAULT_TIMEOUT_MINUTES, datasets
  from tests.integ.timeout import timeout, timeout_and_delete_endpoint_by_name
label_desc: Use image node_id 2 for calling a global function with example usage: training_set() and returns: datasets