torc 0.21.0

Workflow management system
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

#!/usr/bin/env python

"""Test script to run post-processing."""

import json
import shutil
import sys
from pathlib import Path

# command="python postprocess.py -i f2.json -i f3.json -o f4.json",
if len(sys.argv) != 7:
    msg = f"bad inputs in postprocess.py: {sys.argv}"
    raise Exception(msg)

print(f"running {sys.argv}")

# Test that we got a GPU node.
assert shutil.which("nvidia-smi") is not None

input_file1 = Path(sys.argv[2])
input_file2 = Path(sys.argv[4])
output_file = Path(sys.argv[6])
if not input_file1.exists():
    msg = f"{input_file1} does not exist"
    raise Exception(msg)
if not input_file2.exists():
    msg = f"{input_file2} does not exist"
    raise Exception(msg)

output_file.write_text(json.dumps({"hello": "world"}, indent=2) + "\n", encoding="utf-8")