Update the query_by_arch function in API to be compatiable with the submission version of NAS-Bench-201

2020-07-08 05:08:55 +00:00
parent 233a829bd7
commit af1be7f740
3 changed files with 7 additions and 4 deletions
--- a/lib/nas_201_api/api_201.py
+++ b/lib/nas_201_api/api_201.py
@@ -123,7 +123,7 @@ class NASBench201API(NASBenchMetaAPI):
    """
    if self.verbose:
      print('Call query_info_str_by_arch with arch={:} and hp={:}'.format(arch, hp))
-    self._query_info_str_by_arch(arch, hp, print_information)
+    return self._query_info_str_by_arch(arch, hp, print_information)

  # obtain the metric for the `index`-th architecture
  # `dataset` indicates the dataset: