Little fix for NetSpec: allow setting top blob names using string #3747

Merged
merged 1 commit into from Feb 29, 2016
Jump to file or symbol
Failed to load files and symbols.
+6 −0
Split
View
@@ -175,6 +175,12 @@ def __setattr__(self, name, value):
def __getattr__(self, name):
return self.tops[name]
+ def __setitem__(self, key, value):
+ self.__setattr__(key, value)
+
+ def __getitem__(self, item):
+ return self.__getattr__(item)
+
def to_proto(self):
names = {v: k for k, v in six.iteritems(self.tops)}
autonames = Counter()