fetch_ml/tests/fixtures/examples/pyproject_project/train.py

19 lines
446 B
Python

#!/usr/bin/env python3
import argparse
import json
from pathlib import Path
def main() -> None:
parser = argparse.ArgumentParser()
parser.add_argument("--output_dir", required=True)
args = parser.parse_args()
out = Path(args.output_dir)
out.mkdir(parents=True, exist_ok=True)
with (out / "results.json").open("w") as f:
json.dump({"ok": True, "project": "pyproject"}, f)
if __name__ == "__main__":
main()