code: def arguments(self):
      context = load_step_compilation_context()
      args = {
          "Output": f"s3://bucket/{context.pipeline_name}/{context.step_name}"
      }
      if self._input_data:
          args["Input"] = self._input_data
      return args
  
imports: from __future__ import absolute_import
  import pytest
  from sagemaker.workflow.steps import Step, StepTypeEnum
  from sagemaker.workflow.function_step import step
  from sagemaker.workflow.step_collections import StepCollection
  from sagemaker.workflow.condition_step import ConditionStep
  from sagemaker.workflow.conditions import ConditionEquals
  from sagemaker.workflow.properties import Properties
  from sagemaker.workflow.pipeline_definition_config import PipelineDefinitionConfig
  from sagemaker.workflow._steps_compiler import StepsCompiler
  from sagemaker.workflow.utilities import load_step_compilation_context, list_to_request
label_desc: Use image node_id 2 for calling the CustomStep obj's underlying member method code with example usage: obj.arguments() and returns: args