diff --git a/plugin/xprof/integration_tests/tpu/tensorflow/tpu_tf2_keras_test.py b/plugin/xprof/integration_tests/tpu/tensorflow/tpu_tf2_keras_test.py index 57e664f5..fc0b92e3 100644 --- a/plugin/xprof/integration_tests/tpu/tensorflow/tpu_tf2_keras_test.py +++ b/plugin/xprof/integration_tests/tpu/tensorflow/tpu_tf2_keras_test.py @@ -108,6 +108,7 @@ def test_tools_are_in_list(self): 'graph_viewer', 'hlo_stats', 'roofline_model', + 'perf_counters', ] expected.sort() self.assertListEqual(expected, result)