[FIX](handlers) type hint after init_handler

wbenbihi · Aug 25, 2022 · 809aabb · 809aabb
1 parent ec03a0d
commit 809aabb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/hourglass_tensorflow/handlers/train.py b/hourglass_tensorflow/handlers/train.py
@@ -38,6 +38,7 @@ def __init__(
         self._optimizer: Union[HTFObjectReference[Optimizer], str] = None
         self._metrics: List[HTFObjectReference[Metric]] = None
         self._callbacks: List[HTFObjectReference[Callback]] = None
+        self.init_handler()
 
     @property
     def config(self) -> HTFTrainConfig: