code: def test_modify_node():
      fit_calls = (
          "estimator.fit(run_tensorboard_locally=True)",
          "estimator.fit(run_tensorboard_locally=False)",
      )
      modifier = tf_legacy_mode.TensorBoardParameterRemover()
  
      for call in fit_calls:
          node = ast_call(call)
          modifier.modify_node(node)
          assert "estimator.fit()" == pasta.dump(node)
  
imports: from __future__ import absolute_import
  import pasta
  from sagemaker.cli.compatibility.v2.modifiers import tf_legacy_mode
  from tests.unit.sagemaker.cli.compatibility.v2.modifiers.ast_converter import ast_call
label_desc: Use image node_id 4 for calling a global function with example usage: test_modify_node() without return types