Files
quant-trader-service/training/scripts/16_validate_artifact_bundle.py
T

23 lines
604 B
Python

from __future__ import annotations
import argparse
from pathlib import Path
import _bootstrap # noqa: F401
from trader_training.io_utils import setup_logging
from trader_training.validator import validate_artifact_bundle
def main() -> None:
parser = argparse.ArgumentParser()
parser.add_argument("--artifact-root", type=Path, required=True)
parser.add_argument("--require-active", action="store_true")
parser.add_argument("--run-onnx", action="store_true")
args = parser.parse_args()
setup_logging()
validate_artifact_bundle(args)
if __name__ == "__main__":
main()