fix agg

Author: xadupre

_unittests/ut_helpers/data/data-agg.zip

@@ -22,10 +22,10 @@ def df1(cls):
                 textwrap.dedent(
                     """
                     date,version_python,version_transformers,model_name,model_exporter,time_load,time_latency,time_baseline,disc_ort,disc_ort2
-                    2025/01/01,3.13.3,4.52.4,phi3,export,0.5,0.1,0.1,1e-5,1e-5
-                    2025/01/02,3.13.3,4.52.4,phi3,export,0.6,0.11,0.1,1e-5,1e-5
-                    2025/01/01,3.13.3,4.52.4,phi4,export,0.5,0.1,0.105,1e-5,1e-5
-                    2025/01/01,3.12.3,4.52.4,phi4,onnx-dynamo,0.5,0.1,0.999,1e-5,1e-5
+                    2025/01/01,3.13.3,4.52.4,phi3,export,0.51,0.1,0.1,1e-5,1e-5
+                    2025/01/02,3.13.3,4.52.4,phi3,export,0.62,0.11,0.11,1e-5,1e-5
+                    2025/01/01,3.13.3,4.52.4,phi4,export,0.53,0.1,0.105,1e-5,1e-5
+                    2025/01/01,3.12.3,4.52.4,phi4,onnx-dynamo,0.54,0.14,0.999,1e-5,1e-5
                     """
                 )
             )
@@ -99,7 +99,11 @@ def test_cube_logs_view_repr(self):
     def test_cube_logs_view(self):
         cube = self.cube1(verbose=1)
         view = cube.view(
-            CubeViewDef(["version.*", "model_name"], ["time_latency", "time_baseline"])
+            CubeViewDef(
+                ["version.*", "model_name"],
+                ["time_latency", "time_baseline"],
+                ignore_columns=["date"],
+            )
         )
         self.assertEqual((3, 4), view.shape)
         self.assertEqual(
@@ -117,7 +121,10 @@ def test_cube_logs_view(self):
 
         view = cube.view(
             CubeViewDef(
-                ["version.*"], ["time_latency", "time_baseline"], order=["model_exporter"]
+                ["version.*"],
+                ["time_latency", "time_baseline"],
+                order=["model_exporter"],
+                ignore_columns=["date"],
             )
         )
         self.assertEqual((2, 6), view.shape)
@@ -140,12 +147,13 @@ def test_cube_logs_view_agg(self):
             CubeViewDef(
                 ["version.*", "model.*"],
                 ["time_latency", "time_baseline"],
-                key_agg=["model_name"],
+                key_agg=["model_name", "date"],
+                ignore_columns=["version_python"],
             )
         )
         self.assertEqual((2, 2), view.shape)
         self.assertEqual(["time_baseline", "time_latency"], list(view.columns))
-        self.assertEqual([("3.13.3", "export"), ("3.12.3", "onnx-dynamo")], list(view.index))
+        self.assertEqual([("export",), ("onnx-dynamo",)], list(view.index))
 
     @hide_stdout()
     def test_cube_logs_excel(self):
@@ -205,7 +213,7 @@ def test_cube_logs_performance(self):
                 "time",
                 "time_export",
                 "err",
-                "cmd",
+                # "cmd",
                 "bucket-speedup",
                 "raw-short",
             ],

_unittests/ut_helpers/test_log_helper.py

@@ -686,7 +686,7 @@ def _cmd_agg(argv: List[Any]):
 
     csv = list(
         enumerate_csv_files(
-            args.inputs, verbose=args.verbose, filtering=lambda name: reg.search(name)
+            args.inputs, verbose=args.verbose, filtering=lambda name: bool(reg.search(name))
         )
     )
     assert csv, f"No csv files in {args.inputs}"