How to check the value in constant when using pattern matching in onnx-graphsurgeon

[peanutPod]

for example:
onnx model op:

pattern=gs.GraphPattern()
pow_in = pattern.constant()
in_0=pattern.variable()
pow=pattern.add("Node6",op="Pow",inputs=[in_0,pow_in],num_output_tensors=1, check_func=check_constant)

How do I verify that pow_in has a value of 2 when matching

[peanutPod]

@kevinch-nv could you help me?

[peanutPod]

solver:

  def check_constant(node):
      y_value=node.inputs[1].inputs[0].attrs["value"].values
      if y_value.size==1 and y_value==2:
          return True
      return False

  pow_y = pattern.add("ConstantNode", "Constant")

  pow=pattern.add("Node6",op="Pow",inputs=[sub,pow_y],check_func=check_constant,num_output_tensors=1)