Skip to content

Commit a0623ba

Browse files
authored
fix multiple inheritance (aws#191)
1 parent d17e970 commit a0623ba

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

smdebug/tensorflow/keras.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,8 @@ def __init__(
3939
save_all=False,
4040
include_workers="one",
4141
):
42-
super().__init__(
42+
TensorflowBaseHook.__init__(
43+
self,
4344
out_dir=out_dir,
4445
export_tensorboard=export_tensorboard,
4546
tensorboard_dir=tensorboard_dir,
@@ -52,6 +53,7 @@ def __init__(
5253
save_all=save_all,
5354
include_workers=include_workers,
5455
)
56+
tf.keras.callbacks.Callback.__init__(self)
5557
self.tensor_refs_to_save_this_step = set()
5658
self._fetches_added = set()
5759
self.callable_cache = CallableCache()

0 commit comments

Comments
 (0)