code: def test_dataset_name(mnist_visiondata_train, mnist_visiondata_test):
      mnist_visiondata_train.name = "Ref"
      mnist_visiondata_test.name = "Win"
  
      result = HeatmapComparison().run(
          mnist_visiondata_train, mnist_visiondata_test
      )
  
      assert_that(result.display[0].layout.annotations[0].text, "Ref")
      assert_that(result.display[0].layout.annotations[1].text, "Win")
  
imports: from hamcrest import assert_that, close_to, greater_than, has_length
  from deepchecks.vision.checks import HeatmapComparison
label_desc: Use image node_id 5 for calling a global function with example usage: test_dataset_name(mnist_visiondata_train, mnist_visiondata_test) without return types