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