issue/853 - Bug fix: import TestCase and utils class from new file (#854)

* issue/853 - Bug fix: import TestCase and utils class from new file

* issue/853 - import class from framework

test/infinicore/nn/embedding.py

@@ -4,10 +4,15 @@ import sys
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
 
 import torch
-from framework.base import BaseOperatorTest, TensorSpec, TestCase
-from framework.runner import GenericTestRunner
-from framework.tensor import TensorInitializer
-from framework.utils import convert_infinicore_to_torch
+
+from framework import (
+    BaseOperatorTest,
+    GenericTestRunner,
+    TensorInitializer,
+    TensorSpec,
+    TestCase,
+    convert_infinicore_to_torch,
+)
 
 import infinicore
 

test/infinicore/nn/linear.py

@@ -4,8 +4,13 @@ import sys
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
 
 import torch
-from framework.base import BaseOperatorTest, TensorSpec, TestCase
-from framework.runner import GenericTestRunner
+
+from framework import (
+    BaseOperatorTest,
+    GenericTestRunner,
+    TensorSpec,
+    TestCase
+)
 
 import infinicore
 

test/infinicore/nn/rmsnorm.py

@@ -4,8 +4,13 @@ import sys
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
 
 import torch
-from framework.base import BaseOperatorTest, TensorSpec, TestCase
-from framework.runner import GenericTestRunner
+
+from framework import (
+    BaseOperatorTest,
+    GenericTestRunner,
+    TensorSpec,
+    TestCase
+)
 
 import infinicore
 

test/infinicore/nn/rope.py

@@ -4,8 +4,13 @@ import sys
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
 
 import torch
-from framework.base import BaseOperatorTest, TensorSpec, TestCase
-from framework.runner import GenericTestRunner
+
+from framework import (
+    BaseOperatorTest,
+    GenericTestRunner,
+    TensorSpec,
+    TestCase
+)
 from infinicore.nn.functional import RopeAlgo
 
 import infinicore

test/infinicore/tensor/narrow.py

@@ -5,9 +5,15 @@ sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
 
 import torch
 import infinicore
-from framework.base import BaseOperatorTest, TensorSpec, TestCase
-from framework.runner import GenericTestRunner
-from framework.utils import is_broadcast
+
+from framework import (
+    BaseOperatorTest,
+    GenericTestRunner,
+    is_broadcast,
+    TensorSpec,
+    TestCase
+)
+
 
 # ==============================================================================
 # Operator-specific configuration

test/infinicore/tensor/squeeze.py

@@ -5,9 +5,15 @@ sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
 
 import torch
 import infinicore
-from framework.base import BaseOperatorTest, TensorSpec, TestCase
-from framework.runner import GenericTestRunner
-from framework.utils import is_broadcast
+
+from framework import (
+    BaseOperatorTest,
+    GenericTestRunner,
+    TensorSpec,
+    TestCase,
+    is_broadcast
+)
+
 
 # ==============================================================================
 # Operator-specific configuration

test/infinicore/tensor/unsqueeze.py

@@ -5,9 +5,15 @@ sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
 
 import torch
 import infinicore
-from framework.base import BaseOperatorTest, TensorSpec, TestCase
-from framework.runner import GenericTestRunner
-from framework.utils import is_broadcast
+
+from framework import (
+    BaseOperatorTest,
+    GenericTestRunner,
+    TensorSpec,
+    TestCase,
+    is_broadcast
+)
+
 
 # ==============================================================================
 # Operator-specific configuration