From c88945ed48558167840ee152f385250014765792 Mon Sep 17 00:00:00 2001 From: Azad Free Date: Fri, 19 Apr 2024 12:50:26 -0700 Subject: [PATCH] removing base_module for glow/fb/torch_glow/py/NNPI:torch_glow Differential Revision: D56368043 --- torch_glow/tests/backends/NNPI/model_serialization_test.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/torch_glow/tests/backends/NNPI/model_serialization_test.py b/torch_glow/tests/backends/NNPI/model_serialization_test.py index fe4b3079c..26950f378 100644 --- a/torch_glow/tests/backends/NNPI/model_serialization_test.py +++ b/torch_glow/tests/backends/NNPI/model_serialization_test.py @@ -3,9 +3,10 @@ import unittest +import glow.fb.torch_glow.py.NNPI.torch_glow as torch_glow + import torch import torch.jit -import torch_glow from glow.glow.torch_glow.tests.tests import utils