diff --git a/sdk/edir/et_schema.py b/sdk/edir/et_schema.py index 3f4da234322..487960d8bd5 100644 --- a/sdk/edir/et_schema.py +++ b/sdk/edir/et_schema.py @@ -16,6 +16,7 @@ from collections import defaultdict from enum import Enum +from types import NoneType from typing import Any, Dict, List, Optional, Set, Tuple import torch @@ -147,6 +148,8 @@ def _parse_args( # noqa: C901 raise Exception( f"Unsupported argument encountered in list {arg}, {type(arg[0])}" ) + elif isinstance(arg, NoneType): + continue else: raise Exception(f"Unsupported argument encountered {op}, {name}, {arg}")