code: def task_decorate(
      self,
      step_func,
      flow,
      graph,
      retry_count,
      max_user_code_retries,
      ubf_context,
  ):
      return super().task_decorate(
          step_func,
          flow,
          graph,
          retry_count,
          max_user_code_retries,
          ubf_context,
      )
  
file_path: metaflow/metaflow/plugins/frameworks/pytorch.py
imports: import inspect
  import subprocess
  import pickle
  import tempfile
  import os
  import sys
  from metaflow import current
  from metaflow.plugins.parallel_decorator import ParallelDecorator
label_desc: Use image node_id 1 for calling the PytorchParallelDecorator obj's underlying member method code with example usage: obj.task_decorate(step_func, flow, graph, retry_count, max_user_code_retries, ubf_context) and returns: super