def load_run(cls, dump_file, disable_tracking_api): # type: (FileTarget, bool) -> DatabandRun logger.info("Loading dbnd run from %s", dump_file) with dump_file.open("rb") as fp: databand_run = cloudpickle.load(file=fp) if disable_tracking_api: databand_run.context.tracking_store.disable_tracking_api() logger.info("Tracking has been disabled") try: if databand_run.context.settings.core.pickle_handler: pickle_handler = load_python_callable( databand_run.context.settings.core.pickle_handler) pickle_handler(databand_run) except Exception as e: logger.warning( "error while trying to handle pickle with custom handler:", e) return databand_run
def target_to_value(self, target, **kwargs): h = tensorflow.python.keras.callbacks.History() with target.open("rb") as dumped: h.history = cloudpickle.load(dumped) return h