[FIX] hardcoded budget

franchuterivera · franchuterivera · commit f8f697f21826 · 2021-04-19T21:26:06.000+02:00
diff --git a/autoPyTorch/evaluation/tae.py b/autoPyTorch/evaluation/tae.py
@@ -199,7 +199,7 @@ def run_wrapper(
                 )
         else:
             if run_info.budget == 0:
-                run_info = run_info._replace(budget=100.0)
+                run_info = run_info._replace(budget=self.pipeline_config[self.budget_type])
             elif run_info.budget <= 0 or run_info.budget > 100:
                 raise ValueError('Illegal value for budget, must be >0 and <=100, but is %f' %
                                  run_info.budget)

Original file line number	Diff line number	Diff line change
`@@ -199,7 +199,7 @@ def run_wrapper(`
`199`	`199`	`)`
`200`	`200`	`else:`
`201`	`201`	`if run_info.budget == 0:`
`202`		`- run_info = run_info._replace(budget=100.0)`
	`202`	`+ run_info = run_info._replace(budget=self.pipeline_config[self.budget_type])`
`203`	`203`	`elif run_info.budget <= 0 or run_info.budget > 100:`
`204`	`204`	`raise ValueError('Illegal value for budget, must be >0 and <=100, but is %f' %`
`205`	`205`	`run_info.budget)`