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