feat: ✨ add det_predictor/reco_predictor arch in OCRPredictor __repr__ (

#595)
mindee · Nov 9, 2021 · 736222f · 736222f
1 parent 0064a9b
commit 736222f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/doctr/models/predictor/tensorflow.py b/doctr/models/predictor/tensorflow.py
@@ -26,6 +26,7 @@ class OCRPredictor(NestedObject, _OCRPredictor):
         det_predictor: detection module
         reco_predictor: recognition module
     """
+    _children_names = ['det_predictor', 'reco_predictor']
 
     def __init__(
         self,