FIX: consistent origin information

xiaoxiao19 · Nov 22, 2017 · 6e4082c · 6e4082c
1 parent a944e76
commit 6e4082c
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 2 deletions.
diff --git a/smac/initial_design/default_configuration_design.py b/smac/initial_design/default_configuration_design.py
@@ -58,4 +58,6 @@ def _select_configuration(self):
         config: Configuration
             Initial incumbent configuration.
         """
-        return self.scenario.cs.get_default_configuration()
+        config = self.scenario.cs.get_default_configuration()
+        config.origin = 'Default'
+        return config
diff --git a/smac/initial_design/multi_config_initial_design.py b/smac/initial_design/multi_config_initial_design.py
@@ -89,6 +89,9 @@ def run(self) -> Configuration:
             Initial incumbent configuration
         """
         configs = self.configs
+        for config in configs:
+            if config.origin is None:
+                config.origin = 'Initial design'
 
         self.traj_logger.add_entry(train_perf=2**31,
                                    incumbent_id=1,

diff --git a/smac/initial_design/random_configuration_design.py b/smac/initial_design/random_configuration_design.py
@@ -58,4 +58,6 @@ def _select_configuration(self):
         config: Configuration()
             Initial incumbent configuration
         """
-        return self.scenario.cs.sample_configuration()
+        config = self.scenario.cs.sample_configuration()
+        config.origin = 'Random initial design.'
+        return config
diff --git a/smac/initial_design/single_config_initial_design.py b/smac/initial_design/single_config_initial_design.py
@@ -61,6 +61,8 @@ def run(self):
             Initial incumbent configuration
         """
         initial_incumbent = self._select_configuration()
+        if initial_incumbent.origin is None:
+            initial_incumbent.origin = 'Initial design'
 
         # add this incumbent right away to have an entry to time point 0
         self.traj_logger.add_entry(train_perf=2**31,