corrected type hints, added a safety assert

pavel-esir · Sep 21, 2023 · 4d91f34 · 4d91f34
1 parent 78631a4
commit 4d91f34
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/bindings/python/src/openvino/frontend/tensorflow/utils.py b/src/bindings/python/src/openvino/frontend/tensorflow/utils.py
@@ -269,6 +269,9 @@ def tf_function(args):
     def are_shapes_defined(shape: Union[List, Dict]):
         if shape is None:
             return False
+        assert hasattr(shape, '__len__')
+        if len(shape) == 0:
+            return False
 
         if isinstance(shape, list):
             return np.all([shape is not None for shape in input_shapes])

diff --git a/tools/ovc/openvino/tools/ovc/cli_parser.py b/tools/ovc/openvino/tools/ovc/cli_parser.py
@@ -37,7 +37,7 @@ def is_shape_type(value):
         return True
     return False
 
-def single_input_to_input_cut_info(input: [tuple, list, PartialShape, Type, type]):
+def single_input_to_input_cut_info(input: [str, tuple, list, PartialShape, Type, type]):
     """
     Parses parameters of single input to InputCutInfo.
     :param input: input cut parameters of single input