wandb

1
#!/usr/bin/env python
2
"""Test for WB-3758."""
3

4
import os
5
import sys
6

7
import wandb
8

9

10
def main(argv):
11
    # test to ensure
12
    run = wandb.init()
13
    run_project = run.project
14
    run_id = run.id
15
    print(f"Started run {run_project}/{run_id}")
16

17
    try:
18
        os.makedirs("./chdir_test")
19
    except Exception as e:
20
        print(f"Failed to create directory: {e}")
21
        pass
22

23
    os.chdir("./chdir_test")
24
    # log some table data, which is saved in the media folder
25
    pr_data = [
26
        ["setosa", 1.0, 1.0],
27
        ["setosa", 1.0, 1.0],
28
        ["setosa", 1.0, 1.0],
29
        ["setosa", 1.0, 1.0],
30
        ["setosa", 1.0, 1.0],
31
        ["setosa", 1.0, 1.0],
32
        ["setosa", 1.0, 1.0],
33
        ["setosa", 1.0, 1.0],
34
        ["setosa", 1.0, 1.0],
35
        ["setosa", 1.0, 1.0],
36
        ["setosa", 1.0, 0.0],
37
        ["setosa", 1.0, 0.0],
38
        ["setosa", 1.0, 0.0],
39
        ["setosa", 1.0, 0.0],
40
        ["setosa", 1.0, 0.0],
41
        ["setosa", 1.0, 0.0],
42
        ["setosa", 1.0, 0.0],
43
        ["setosa", 1.0, 0.0],
44
        ["setosa", 1.0, 0.0],
45
        ["setosa", 1.0, 0.0],
46
        ["versicolor", 1.0, 1.0],
47
        ["versicolor", 1.0, 1.0],
48
        ["versicolor", 1.0, 1.0],
49
        ["versicolor", 1.0, 1.0],
50
        ["versicolor", 1.0, 1.0],
51
        ["versicolor", 1.0, 1.0],
52
        ["versicolor", 1.0, 1.0],
53
        ["versicolor", 1.0, 1.0],
54
        ["versicolor", 1.0, 1.0],
55
        ["versicolor", 1.0, 1.0],
56
        ["versicolor", 1.0, 0.0],
57
        ["versicolor", 1.0, 0.0],
58
        ["versicolor", 1.0, 0.0],
59
        ["versicolor", 1.0, 0.0],
60
        ["versicolor", 1.0, 0.0],
61
        ["versicolor", 1.0, 0.0],
62
        ["versicolor", 1.0, 0.0],
63
        ["versicolor", 1.0, 0.0],
64
        ["versicolor", 1.0, 0.0],
65
        ["versicolor", 1.0, 0.0],
66
    ]
67

68
    # convert the data to a table
69
    pr_table = wandb.Table(data=pr_data, columns=["class", "precision", "recall"])
70
    wandb.log({"pr_table": pr_table})
71
    wandb.finish()
72

73
    # Check results
74
    api = wandb.Api()
75
    last_run = api.run(f"{run_project}/{run_id}")
76
    media_path = last_run.summary_metrics["pr_table"]["path"]
77
    media_file = last_run.file(media_path)
78
    assert media_file.size > 0
79
    print("Success")
80

81

82
if __name__ == "__main__":
83
    main(sys.argv)
84