NetSpec: allow setting blob names by string

This commit is contained in:
Anatoly Baksheev 2016-02-29 18:01:07 +03:00
Родитель c2769c1096
Коммит cb277769a4
1 изменённых файлов: 6 добавлений и 0 удалений

Просмотреть файл

@ -175,6 +175,12 @@ class NetSpec(object):
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()