Tag coiled.function in TPC-H (#1540)

coiled · Sep 5, 2024 · 0a9b5d8 · 0a9b5d8
1 parent a11f90d
commit 0a9b5d8
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/tpch/conftest.py b/tests/tpch/conftest.py
@@ -315,7 +315,7 @@ def machine_spec(scale):
 
 
 @pytest.fixture(scope="module")
-def module_run(local, module, scale, name, machine_spec):
+def module_run(local, module, scale, name, machine_spec, github_cluster_tags):
     if local:
 
         def _run(function):
@@ -325,7 +325,11 @@ def _run(function):
 
     else:
 
-        @coiled.function(**machine_spec, name=f"tpch-{module}-{scale}-{name}")
+        @coiled.function(
+            **machine_spec,
+            name=f"tpch-{module}-{scale}-{name}",
+            tags=github_cluster_tags,
+        )
         def _run(function):
             return function()