Dont unpatch keras
@@ -506,14 +506,14 @@ def uninit():
run = config = None
watch_called = False
# UNDO patches
- for mod in patched["tensorboard"] + patched["keras"]:
+ for mod in patched["tensorboard"]:
module = import_module(mod[0])
parts = mod[1].split(".")
if len(parts) > 1:
module = getattr(module, parts[0])
mod[1] = parts[1]
setattr(module, mod[1], getattr(module, "orig_"+mod[1]))
- patched = {"tensorboard": [], "keras": []}
+ patched["tensorboard"] = []
_saved_files = set()
