Stricter name checking
@@ -41,9 +41,9 @@ VALUE_BYTES_LIMIT = 100000
def fullname(o):
name = o.__class__.__module__.split(".")[0] + "." + o.__class__.__name__
# TODO: There are likely other cases to handle here
- if name.startswith("tensorflow"):
+ if name.startswith("tensorflow."):
name = "tensorflow.Tensor"
- elif name.startswith("torch"):
+ elif name.startswith("torch."):
name = "torch.Tensor"
return name
