demo/constants.py

"""
Constants for the demo.
"""

import os

ONNX_MODEL_DIRECTORY = "demo/onnx-models"
LEELA_MODEL_DIRECTORY = "demo/leela-models"
FIGURE_DIRECTORY = "demo/figures"

ONNX_MODEL_NAMES = [
    f for f in os.listdir(ONNX_MODEL_DIRECTORY)
    if f.endswith(".onnx")
]
LEELA_MODEL_NAMES = [
    f for f in os.listdir(LEELA_MODEL_DIRECTORY)
    if f.endswith(".pb.gz")
]