code: def test_deepchecks_dataset_load():
      dataset = load_dataset(train=True)
      assert_that(dataset, instance_of(VisionData))
  
      dataset = load_dataset(train=False)
      assert_that(dataset, instance_of(VisionData))
  
imports: import numpy
  from hamcrest import assert_that, equal_to, instance_of, is_not
  from deepchecks.vision import VisionData
  from deepchecks.vision.datasets.classification.mnist_tensorflow import load_dataset, load_model
label_desc: Use image node_id 1 for calling a global function with example usage: test_deepchecks_dataset_load() without return types