Fix minimizer tests (#1006)

Summary:
Pull Request resolved: #1006

fix tests failures introduced in D56799587

Reviewed By: hl475, kalpit-meta

Differential Revision: D57166956

fbshipit-source-id: 5c49fe28567d34f1ae2c38454f61fc1480241f2f

fx2ait/fx2ait/tools/ait_minimizer.py

@@ -75,12 +75,12 @@ def __init__(
         self.curr_iter = 0  # We use this counter to prevent duplicate .so naming
         super().__init__(module, sample_input, compare_fn, settings)
 
-    def run_a(self, mod, inputs):
+    def run_a(self, mod, inputs, _report_idx=-1):
         mod.eval()
         with torch.no_grad():
             return mod(*inputs)
 
-    def run_b(self, mod, inputs):
+    def run_b(self, mod, inputs, _report_idx=-1):
         mod.eval()
         dll_name = f"{self.name}_{self.curr_iter}.so"
         self.curr_iter += 1