Fix benchmark experiment tests.

pytorch · Oct 1, 2024 · 87db304 · 87db304
1 parent 4f25209
commit 87db304
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/benchmarks/test_benchmark_experiment.py b/test/benchmarks/test_benchmark_experiment.py
@@ -9,7 +9,7 @@ def test_to_dict(self):
     be = BenchmarkExperiment("cpu", "PJRT", "some xla_flags", "openxla", None,
                              False, "train", "123", False)
     actual = be.to_dict()
-    self.assertEqual(9, len(actual))
+    self.assertEqual(10, len(actual))
     self.assertEqual("cpu", actual["accelerator"])
     self.assertTrue("accelerator_model" in actual)
     self.assertEqual("PJRT", actual["xla"])