add smac_2_10-dev to setup.py

automl · May 4, 2015 · 604c49e · 604c49e
1 parent 9d07a35
commit 604c49e
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -200,7 +200,7 @@ def run(self):
         # COPY/DOWNLOAD OPTIMIZER TO ROOT FOLDER
         optimizer_dir = os.path.join(here_we_are, "optimizers")
 
-        tpe, smac, spearmint = (False, False, False)
+        tpe, smac, spearmint, smac_2_08, smac_2_10 = (False, False, False, False, False)
         tpe = self._copy_and_download_optimizer(optimizer_dir=optimizer_dir,
                                                 optimizer_name='tpe',
                                                 optimizer_tar_name="hyperopt_august2013_mod_src.tar.gz",
@@ -226,26 +226,34 @@ def run(self):
                                                           optimizer_tar_name="smac_2_08_00-master_src.tar.gz",
                                                           url="http://www.automl.org/smac_2_08_00-master_src.tar.gz",
                                                           md5='2be626a5437b56da2eba1b67b7a94367')
+            smac_2_10 = self._copy_and_download_optimizer(optimizer_dir=optimizer_dir,
+                                                          optimizer_name='smac',
+                                                          optimizer_tar_name="smac_2_10_00-dev_src.tar.gz",
+                                                          url="http://www.automl.org/smac_2_10_00-dev_src.tar.gz",
+                                                          md5='510a9ad71e0713bd5ac906f021d8d3be')
 
         # TODO: Normally one wants to call run(self), but this runs distutils and ignores install_requirements for unknown reasons
         # if anyone knows a better way, feel free to change
         install.do_egg_install(self)
 
         # Give detailed output to user
-        if not tpe or not smac or not spearmint or not smac_2_08:
+        if not tpe or not smac or not spearmint or not smac_2_08 or not smac_2_10:
             sys.stderr.write("[ERROR] Something went wrong while copying and downloading optimizers." +
                              "Please do the following to be ready to start optimizing:\n\n" +
                              "cd optimizers\n" +
                              "wget http://www.automl.org/hyperopt_august2013_mod_src.tar.gz \n" +
                              "wget http://www.automl.org/smac_2_06_01-dev_src.tar.gz \n" +
                              "wget http://www.automl.org/smac_2_08_00-master_src.tar.gz \n" +
+                             "wget http://www.automl.org/smac_2_10_00-dev_src.tar.gz \n", +
                              "wget http://www.automl.org/spearmint_april2013_mod_src.tar.gz \n" +
                              "tar -xf hyperopt_august2013_mod_src.tar.gz \n" +
                              "mv hyperopt_august2013_mod_src tpe/ \n" +
                              "tar -xf smac_2_06_01-dev_src.tar.gz \n" +
-                             "mv smac_2_06_01-dev_src.tar.gz smac/ \n" +
+                             "mv smac_2_06_01-dev_src smac/ \n" +
                              "tar -xf smac_2_08_00-master_src.tar.gz \n" +
-                             "mv smac_2_08_00-master_src.tar.gz smac/ \n" +
+                             "mv smac_2_08_00-master_src smac/ \n" +
+                             "tar -xf smac_2_10_00-dev_src.tar.gz \n" +
+                             "mv smac_2_10_00-dev_src smac/ \n" +
                              "tar -xf spearmint_april2013_mod_src.tar.gz \n" +
                              "mv spearmint_april2013_mod_src spearmint/ \n\n" +
                              "Thank You!\n")