dream

1
import requests
2

3

4
def round_struct(struct, digits):
5
    if isinstance(struct, dict):
6
        return {k: round_struct(v, digits) for k, v in struct.items()}
7
    elif isinstance(struct, list):
8
        return [round_struct(v, digits) for v in struct]
9
    elif isinstance(struct, float):
10
        return round(struct, digits)
11
    else:
12
        return struct
13

14

15
def test_respond():
16
    url = "http://0.0.0.0:8088/respond"
17

18
    text = ["Hello, it's [MASK] dog from my [MASK]."]
19

20
    request_data = {"text": text}
21

22
    result = requests.post(url, json=request_data).json()
23

24
    gold_result = {
25
        "predicted_tokens": [
26
            [
27
                {
28
                    "a": 0.21459759771823883,
29
                    "another": 0.0011624041944742203,
30
                    "his": 0.0008239049348048866,
31
                    "my": 0.05392618849873543,
32
                    "our": 0.0016213968629017472,
33
                    "some": 0.0008065433939918876,
34
                    "that": 0.014674700796604156,
35
                    "the": 0.6869651079177856,
36
                    "this": 0.002329436829313636,
37
                    "your": 0.015044458210468292,
38
                },
39
                {
40
                    "apartment": 0.03887254372239113,
41
                    "childhood": 0.015000063925981522,
42
                    "class": 0.01927136816084385,
43
                    "dream": 0.036867350339889526,
44
                    "dreams": 0.04079044237732887,
45
                    "house": 0.10746614634990692,
46
                    "neighborhood": 0.03330889344215393,
47
                    "office": 0.03672104701399803,
48
                    "room": 0.042806293815374374,
49
                    "world": 0.015730833634734154,
50
                },
51
            ]
52
        ]
53
    }
54
    digits = 2
55
    result = round_struct(result, digits)
56
    gold_result = round_struct(gold_result, digits)
57
    assert result == gold_result, f"Got\n{result}\n, but expected:\n{gold_result}"
58
    print("Success")
59

60

61
if __name__ == "__main__":
62
    test_respond()
63