assert_in("Failed saving periodic pipeline, exception", our_file.read())
// clean up
for f in os.listdir("./"):
if search("pipeline_", f):
os.remove(os.path.join("./", f))
def test_save_periodic_pipeline_2():
Assert that _save_periodic_pipeline creates the checkpoint folder and exports to it if it didn"t exist
tpot_obj = TPOTClassifier(
random_state=42,
After Change
tpot_obj._last_pipeline_write = datetime.now()
sleep(0.11)
tpot_obj._output_best_pipeline_period_seconds = 0.1
tmpdir = mkdtemp() + "/"
tpot_obj.periodic_checkpoint_folder = tmpdir
// reset _pareto_front to rasie exception
tpot_obj._pareto_front = None
tpot_obj._save_periodic_pipeline(1)
our_file.seek(0)
assert_in("Failed saving periodic pipeline, exception", our_file.read())
//clean up
rmtree(tmpdir)
def test_save_periodic_pipeline_2():
Assert that _save_periodic_pipeline creates the checkpoint folder and exports to it if it didn"t exist
