code: def __init__(self, base_trial, threshold=10.0):
      super().__init__(base_trial)
      self.threshold = float(threshold)
  
imports: from smdebug.rules.rule import Rule
label_desc: Use image node_id 1 to create a new CustomGradientRule object from inherited base classes: Rule with example: obj = CustomGradientRule(base_trial, threshold)