code: def test_reduce_with_drift_classification(
      mnist_visiondata_train, mnist_visiondata_test
  ):
      # Arrange
      check = PredictionDrift(categorical_drift_method="PSI")
      # Act
      result = check.run(mnist_visiondata_train, mnist_visiondata_test)
      # Assert
      assert_that(
          result.reduce_output(),
          has_entries({"Samples Per Class": close_to(0.057, 0.01)}),
      )
  
file_path: deepchecks/tests/vision/checks/model_evaluation/prediction_drift_test.py
imports: from hamcrest import assert_that, calling, close_to, equal_to, greater_than, has_entries, has_length, raises
  from deepchecks.core.errors import NotEnoughSamplesError
  from deepchecks.vision.checks import PredictionDrift
  from tests.base.utils import equal_condition_result
label_desc: Use image node_id 6 for calling a global function with example usage: test_reduce_with_drift_classification(mnist_visiondata_train, mnist_visiondata_test) without return types