Skip to content

Commit d5fd179

Browse files
committed
Separate model and feature support matrices by category (#1100)
Signed-off-by: Teresa Chen <boe20211@gmail.com>
1 parent 864e9ba commit d5fd179

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

.buildkite/pipeline_generation/add_feature_to_ci.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,8 @@ class FeatureCategory(str, Enum):
1313
FEATURE_SUPPORT = "feature support matrix"
1414
KERNEL_SUPPORT = "kernel support matrix"
1515

16-
def generate_from_template(feature_name: str,
17-
feature_category: str, queue: str) -> None:
16+
def generate_from_template(feature_name: str, feature_category: str,
17+
queue: str) -> None:
1818
"""
1919
Generates a buildkite yml file from feature template.
2020
Args:
@@ -99,7 +99,8 @@ def main():
9999
parser.add_argument(
100100
'--category',
101101
choices=[
102-
FeatureCategory.FEATURE_SUPPORT.value, FeatureCategory.KERNEL_SUPPORT.value
102+
FeatureCategory.FEATURE_SUPPORT.value,
103+
FeatureCategory.KERNEL_SUPPORT.value
103104
],
104105
default='feature support matrix',
105106
help='[OPTIONAL] Category of feature. (Default: feature support matrix)'

.buildkite/pipeline_generation/add_model_to_ci.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ class ModelCategory(str, Enum):
2323
}
2424

2525

26-
def generate_from_template(model_name: str, queue: str, model_type: str,
26+
def generate_from_template(model_name: str, queue: str, model_type: str,
2727
model_category: str) -> None:
2828
"""
2929
Generates a buildkite yml file from model template.

0 commit comments

Comments
 (0)