scalability framework: fix default workload selection (#22588)

MaterializeInc · Oct 23, 2023 · aff5730 · aff5730
1 parent 5843753
commit aff5730
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/test/scalability/mzcompose.py b/test/scalability/mzcompose.py
@@ -343,7 +343,11 @@ def workflow_default(c: Composition, parser: WorkflowArgumentParser) -> None:
     workloads = (
         [globals()[workload] for workload in args.workload]
         if args.workload
-        else [w for w in all_subclasses(Workload) if not w == WorkloadSelfTest]
+        else [
+            workload_cls
+            for workload_cls in all_subclasses(Workload)
+            if not issubclass(workload_cls, WorkloadSelfTest)
+        ]
     )
 
     schema = Schema(