Separate model and feature support matrices by category (#1100)

Signed-off-by: Teresa Chen <boe20211@gmail.com>

Author: boe20211

.buildkite/pipeline_generation/add_feature_to_ci.py

@@ -13,7 +13,8 @@ class FeatureCategory(str, Enum):
     FEATURE_SUPPORT = "feature support matrix"
     KERNEL_SUPPORT = "kernel support matrix"
 
-def generate_from_template(feature_name: str, feature_category: str, queue: str) -> None:
+def generate_from_template(feature_name: str, 
+                           feature_category: str, queue: str) -> None:
     """
     Generates a buildkite yml file from feature template.
     Args:
@@ -97,10 +98,11 @@ def main():
         help="The name of the agent queue to use (ex: 'tpu_v6e_queue')")
     parser.add_argument(
         '--category',
-        choices=[FeatureCategory.FEATURE_SUPPORT.value, FeatureCategory.KERNEL_SUPPORT.value],
+        choices=[
+            FeatureCategory.FEATURE_SUPPORT.value, FeatureCategory.KERNEL_SUPPORT.value
+        ],
         default='feature support matrix',
-        help=
-        '[OPTIONAL] Category of feature. (Default: feature support matrix)'
+        help='[OPTIONAL] Category of feature. (Default: feature support matrix)'
     )
     args = parser.parse_args()
     generate_from_template(feature_name=args.feature_name,

.buildkite/pipeline_generation/add_model_to_ci.py

@@ -23,8 +23,8 @@ class ModelCategory(str, Enum):
 }
 
 
-def generate_from_template(model_name: str, queue: str,
-                           model_type: str, model_category: str) -> None:
+def generate_from_template(model_name: str, queue: str, model_type: str, 
+                           model_category: str) -> None:
     """
     Generates a buildkite yml file from model template.
     Args:
@@ -119,7 +119,9 @@ def main():
     )
     parser.add_argument(
         '--category',
-        choices=[ModelCategory.TEXT_ONLY.value, ModelCategory.MULTIMODEL.value],
+        choices=[
+            ModelCategory.TEXT_ONLY.value, ModelCategory.MULTIMODEL.value
+        ],
         default='text-only',
         help=
         '[OPTIONAL] Category of model. Must be "text-only" or "multimodel". (Default: text-only)'