Update bigbench/benchmark_tasks/simple_arithmetic/test.py
Update bigbench/benchmark_tasks/simple_arithmetic/test.py

Co-authored-by: Jaehoon Lee <09e823099c9eedc323f9e4b76a9c6dc3c4faadc6@google.com>
google/BIG-bench,google/BIG-bench