update code styles

2020-01-09 22:26:23 +11:00
parent 5ac5060a33
commit ad34af9913
26 changed files with 192 additions and 81 deletions
--- a/lib/nas_102_api/api.py
+++ b/lib/nas_102_api/api.py
@@ -70,6 +70,9 @@ class NASBench102API(object):
  def __repr__(self):
    return ('{name}({num}/{total} architectures)'.format(name=self.__class__.__name__, num=len(self.evaluated_indexes), total=len(self.meta_archs)))

+  def random(self):
+    return random.randint(0, len(self.meta_archs)-1)
+
  def query_index_by_arch(self, arch):
    if isinstance(arch, str):
      if arch in self.archstr2index: arch_index = self.archstr2index[ arch ]