wandb

1
import threading
2

3
import wandb
4

5

6
# Checks if wandb has issues during set up in a multithreaded environment
7
def thread_test(n):
8
    run = wandb.init(project="threadtest")
9
    run.log({"thread": n})
10

11

12
def main():
13
    try:
14
        threads = []
15
        for i in range(2):
16
            threads.append(threading.Thread(target=thread_test, args=(i,)))
17
        for thread in threads:
18
            thread.start()
19
        for thread in threads:
20
            thread.join()
21
    except Exception as e:
22
        print(e)
23
        print("Issue with calling wandb init in a multithreaded situation")
24
        raise AssertionError(
25
            "Issue with calling wandb init in a multithreaded situation"
26
        )
27

28

29
if __name__ == "__main__":
30
    main()
31