code: def set_required_tensors(self, step):
      for tname in self.base_trial.tensor_names(collection="gradients"):
          self.req_tensors.add(tname, steps=[step])
  
imports: from smdebug.rules.rule import Rule
label_desc: Use image node_id 2 for calling the CustomGradientRule obj's underlying member method code with example usage: obj.set_required_tensors(step) without return types