fix #5

autoai-org · Oct 4, 2018 · 51e4c30 · 51e4c30
1 parent a461432
commit 51e4c30
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/cvpm/solver.py b/cvpm/solver.py
@@ -37,8 +37,9 @@ def help_message(self):
     def _prepare_models(self, toml_file):
         parsed_toml = toml.load(toml_file)
         downloader = Downloader()
-        for each in parsed_toml["models"]:
-            downloader.download(each["url"], "pretrained")
+        if "models" in parsed_toml.keys():
+            for each in parsed_toml["models"]:
+                downloader.download(each["url"], "pretrained")
 
     def set_ready(self):
         self._isReady = True