From 21726065a6809f8e5385e03717c1d073a1d1db3c Mon Sep 17 00:00:00 2001 From: Katherine Yang <80359429+jbkyang-nvi@users.noreply.github.com> Date: Mon, 16 Oct 2023 19:04:12 -0700 Subject: [PATCH] Fix L0_implicit_state (#6427) (#6442) Co-authored-by: Iman Tabrizian --- qa/common/gen_common.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/qa/common/gen_common.py b/qa/common/gen_common.py index 8bd97720b5..d574627dfd 100644 --- a/qa/common/gen_common.py +++ b/qa/common/gen_common.py @@ -24,6 +24,8 @@ # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. +from typing import List + # Common utilities for model generation scripts import numpy as np @@ -154,5 +156,5 @@ def np_to_torch_dtype(np_dtype): elif np_dtype == np.float64: return torch.double elif np_dtype == np_dtype_string: - return None # Not supported in Torch + return List[str] return None