def _load_orion(pipeline, hyperparameters=None):
if pipeline is None:
return Orion()
elif isinstance(pipeline, Orion):
return pipeline
else:
hyperparameters = _load_dict(hyperparameters)
try:
return Orion(pipeline, hyperparameters)
except ValueError:
try:
return Orion.load(pipeline)
except (FileNotFoundError, UnpicklingError):
raise ValueError('Invalid pipeline: {}'.format(pipeline))
def test_save_load(self, tmpdir):
path = os.path.join(tmpdir, 'some/path.pkl')
self.orion.save(path)
new_orion = Orion.load(path)
assert new_orion == self.orion