зеркало из https://github.com/mozilla/bugbug.git
40 строки
1002 B
Python
40 строки
1002 B
Python
# -*- coding: utf-8 -*-
|
|
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
|
# You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
import logging
|
|
import os
|
|
import re
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def get_model_name() -> str | None:
|
|
pr_description = os.environ.get("PR_DESCRIPTION")
|
|
if not pr_description:
|
|
logger.error("The PR_DESCRIPTION environment variable does not exist")
|
|
return None
|
|
|
|
match = re.search(r"Train on Taskcluster:\s+([a-z_1-9]+)", pr_description)
|
|
if not match:
|
|
logger.error(
|
|
"Could not identify the model name using the 'Train on Taskcluster' keyword from the Pull Request description"
|
|
)
|
|
return None
|
|
|
|
model_name = match.group(1)
|
|
|
|
return model_name
|
|
|
|
|
|
def main():
|
|
model = get_model_name()
|
|
if model:
|
|
print(model)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|