Skip to content

Commit 2fb1fb7

Browse files
authored
Bugfix related to random states and punisher
1 parent 66fc079 commit 2fb1fb7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

autoPyTorch/core/worker_no_timelimit.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ def compute(self, config, budget, working_directory, config_id, **kwargs):
5757
import pynisher
5858

5959
limit_train = pynisher.enforce_limits(mem_in_mb=self.pipeline_config['memory_limit_mb'], wall_time_in_s=int(budget * 4))(self.optimize_pipeline)
60-
result = limit_train(config, budget, config_id, start_time)
60+
result = limit_train(config, budget, config_id, random.getstate())
6161

6262
if (limit_train.exit_status == pynisher.TimeoutException):
6363
raise Exception("Time limit reached. Took " + str((time.time()-start_time)) + " seconds with budget " + str(budget))

0 commit comments

Comments
 (0)