code: def test_import_from_node_should_be_modified_training_input():
      modifier = training_input.ShuffleConfigImportFromRenamer()
      node = ast_import("from sagemaker.session import ShuffleConfig")
      assert modifier.node_should_be_modified(node)
  
file_path: sagemaker-python-sdk/tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_shuffle_config.py
imports: from __future__ import absolute_import
  import pasta
  import pytest
  from sagemaker.cli.compatibility.v2.modifiers import training_input
  from tests.unit.sagemaker.cli.compatibility.v2.modifiers.ast_converter import ast_call, ast_import
label_desc: Use image node_id 6 for calling a global function with example usage: test_import_from_node_should_be_modified_training_input() without return types