fix broken test

facebookresearch · Feb 9, 2023 · d3a3c78 · d3a3c78
1 parent 23db900
commit d3a3c78
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/hta/common/trace.py b/hta/common/trace.py
@@ -637,10 +637,10 @@ def convert_time_series_to_events(
 
         Returns a list of json events that can be appended to the trace.
         """
-        required_columns = ["tid", "pid", "ts", counter_col]
+        required_columns = ["pid", "ts", counter_col]
         if not set(required_columns).issubset(series.columns):
             logger.warning(
-                "Time seried dataframe does NOT contain required columns "
+                "Time series dataframe does NOT contain required columns "
                 f"{required_columns}, columns contained = {series.columns}"
             )
             return []

diff --git a/tests/test_trace_analysis.py b/tests/test_trace_analysis.py
@@ -206,7 +206,7 @@ def test_generate_trace_with_counters(self, mock_write_trace):
 
         counter_events = [ev for ev in trace_json["traceEvents"] if ev["ph"] == PHASE_COUNTER]
         print(f"Trace has {len(counter_events)} counter events")
-        self.assertGreaterEqual(len(counter_events), 23000)
+        self.assertGreaterEqual(len(counter_events), 21000)
 
         counter_names = {ev["name"] for ev in counter_events}
         self.assertEqual(