pytorch

1
#!/usr/bin/env python3
2

3
import os
4
import sys
5

6
import yaml
7

8
# Need to import modules that lie on an upward-relative path
9
sys.path.append(os.path.join(sys.path[0], ".."))
10

11
import cimodel.lib.miniyaml as miniyaml
12

13

14
def regurgitate(depth, use_pyyaml_formatter=False):
15
    data = yaml.safe_load(sys.stdin)
16

17
    if use_pyyaml_formatter:
18
        output = yaml.dump(data, sort_keys=True)
19
        sys.stdout.write(output)
20
    else:
21
        miniyaml.render(sys.stdout, data, depth)
22

23

24
if __name__ == "__main__":
25
    regurgitate(3)
26