File size: 628 Bytes
9c6594c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import google.protobuf

protobuf_version = google.protobuf.__version__[0]

if protobuf_version == "3":
    from wandb.proto.v3.wandb_internal_pb2 import *
elif protobuf_version == "4":
    from wandb.proto.v4.wandb_internal_pb2 import *
elif protobuf_version == "5":
    from wandb.proto.v5.wandb_internal_pb2 import *
elif protobuf_version == "6":
    from wandb.proto.v6.wandb_internal_pb2 import *
else:
    raise ImportError(
        "Failed to import protobufs for protobuf version"
        f" {google.protobuf.__version__}. `wandb` only works with major"
        " versions 3, 4, 5, and 6 of the protobuf package.",
    )