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.",
)
|