Welcome to mirror list, hosted at ThFree Co, Russian Federation.

subprocess_script.py « inputs « decoder « tests - github.com/marian-nmt/marian-regression-tests.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: c17e0c41c1546ff9ca4be5701838fc8005672438 (plain)
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
#!/usr/bin/env python
# -*- coding: utf-8 -*-

import sys
from subprocess import Popen, PIPE


marian_path = sys.argv[1]
config_file = sys.argv[2]
text_file = sys.argv[3]


cmd = ["{}/marian-decoder".format(marian_path), "-c", config_file, "--log", "subprocess.log"]
sys.stdout.write("Command: {}\n".format(cmd))
p = Popen(cmd, stdin=PIPE, stdout=PIPE, stderr=PIPE)


with open(text_file) as f:
    text = f.readlines()[:10]

for sent in text:
    sys.stdout.write("Sending:   {}".format(sent))
    p.stdin.write(sent)
    p.stdin.flush()
    out = p.stdout.readline()
    sys.stdout.write("Receiving: {}".format(out))

p.stdin.close()
p.terminate()