test_owner.py

# Copyright 2023 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""The file of test owners."""

import enum


class Team(enum.Enum):
  SOLUTIONS_TEAM = "solutions_team"
  PYTORCH_XLA = "pytorch_xla"
  MULTIPOD = "multipod"
  MLCOMPASS = "mlcompass"
  INFERENCE = "inference"
  FRAMEWORK = "framework3p"
  LLM_DEVX = "llm_devx"
  SPARSITY_DIFFUSION_DEVX = "sparsity_diffusion_devx"
  PERFORMANCE = "performance"
  PRODUCTIVITY = "productivity"


# XLML - TensorFlow
ERIC_L = "Eric L."
CHANDRA_D = "Chandra D."
GAGIK_A = "Gagik A."

# PYTORCH
PEI_Z = "Pei Z."
MANFEI_B = "Manfei B."

# MaxText
TONY_C = "Tony C."
JON_B = "Jon B."
RAYMOND_Z = "Raymond Z."
MATT_D = "Matt D."
PRIYANKA_G = "Priyanka G."
SURBHI_J = "Surbhi J."
ZHIYU_L = "Zhiyu L."
MOHIT_K = "Mohit K."
ANISHA_M = "Anisha M."
YUWEI_Y = "Yuwei Y."

# Multi-tier Checkpointing
ABHINAV_S = "ABHINAV S."
XUEFENG_G = "XUEFENG G."

# MLCompass
ORTI_B = "Orti B."

# Sparsity & Diffusion DevX
RAN_R = "Ran R."
PARAM_B = "Param B."
MICHELLE_Y = "Michelle Y."

# Inference
ANDY_Y = "Andy Y."
XIANG_S = "Xiang S."
MORGAN_D = "Morgan D."
YIJIA_J = "Yijia J."
PATE_M = "Pate M."

# 3P Ecosystems
RICHARD_L = "Richard L."
WENXIN_D = "Wenxin D."

# FRAMEWORK
QINY_Y = "Qinyi Y."

# JAX
AKANKSHA_G = "Akanksha G."

# MAP_REPRODUCIBILITY
GUNJAN_J = "Gunjan J."

# Bite
Maggie_Z = "Maggie Z."