Skip to content

Commit 6f4bb76

Browse files
committed
more refactors
1 parent 68a358c commit 6f4bb76

File tree

7 files changed

+52
-64
lines changed

7 files changed

+52
-64
lines changed

ivy/transpiler/transformations/transformers/globals_transformer/base_transformer.py

Lines changed: 11 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1,21 +1,19 @@
1-
# global
21
from copy import copy, deepcopy
32
import gast
43
import types
54
from typing import List, Tuple, Optional, Union, TYPE_CHECKING
65

7-
# local
8-
from ..base_transformer import (
6+
from ivy.transpiler.transformations.transformers.base_transformer import (
97
BaseTransformer,
108
)
11-
from ...transformer import Transformer
12-
from ....translations.data.global_like import (
9+
from ivy.transpiler.transformations.transformer import Transformer
10+
from ivy.transpiler.translations.data.global_like import (
1311
GlobalObjectLike,
1412
Position,
1513
StackObjectLike,
1614
)
17-
from ....translations.data.object_like import BaseObjectLike
18-
from ....utils.ast_utils import (
15+
from ivy.transpiler.translations.data.object_like import BaseObjectLike
16+
from ivy.transpiler.utils.ast_utils import (
1917
FRONTEND_STANDARD_GLOBALS_TARGET_TO_MODULE,
2018
IVY_STANDARD_GLOBALS_TARGET_TO_MODULE,
2119
ast_to_source_code,
@@ -27,21 +25,21 @@
2725
FileNameStrategy,
2826
TranslatedContext,
2927
)
30-
from ....utils.api_utils import (
28+
from ivy.transpiler.utils.api_utils import (
3129
get_function_from_modules,
3230
SUPPORTED_BACKENDS_PREFIX,
3331
)
34-
from ....utils.naming_utils import NAME_GENERATOR
35-
from ....utils.origin_utils import ORIGI_INFO
36-
from ....utils.conversion_utils import (
32+
from ivy.transpiler.utils.naming_utils import NAME_GENERATOR
33+
from ivy.transpiler.utils.origin_utils import ORIGI_INFO
34+
from ivy.transpiler.utils.conversion_utils import (
3735
BUILTIN_LIKELY_MODULE_NAMES,
3836
)
39-
from ...configurations.base_transformer_config import (
37+
from ivy.transpiler.transformations.configurations.base_transformer_config import (
4038
BaseTransformerConfig,
4139
)
4240

4341
if TYPE_CHECKING:
44-
from ....translations.data.object_like import (
42+
from ivy.transpiler.translations.data.object_like import (
4543
TypeObjectLike,
4644
FuncObjectLike,
4745
)

ivy/transpiler/transformations/transformers/inject_transformer/base_transformer.py

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,23 +1,21 @@
1-
# global
21
import gast
32
import inspect
43
import textwrap
54

6-
# local
7-
from ...configurations.base_transformer_config import (
5+
from ivy.transpiler.transformations.configurations.base_transformer_config import (
86
BaseTransformerConfig,
97
)
10-
from ...transformer import Transformer
11-
from ....utils import pickling_utils
12-
from ....utils.ast_utils import (
8+
from ivy.transpiler.transformations.transformer import Transformer
9+
from ivy.transpiler.utils import pickling_utils
10+
from ivy.transpiler.utils.ast_utils import (
1311
ast_to_source_code,
1412
is_super_call_node,
1513
)
16-
from ....utils.origin_utils import ORIGI_INFO
17-
from ..base_transformer import (
14+
from ivy.transpiler.utils.origin_utils import ORIGI_INFO
15+
from ivy.transpiler.transformations.transformers.base_transformer import (
1816
BaseTransformer,
1917
)
20-
from ..rename_transformer.base_transformer import (
18+
from ivy.transpiler.transformations.transformers.rename_transformer.base_transformer import (
2119
BaseRenameTransformer,
2220
)
2321

ivy/transpiler/transformations/transformers/inplace_update_transformer/base_transformer.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,21 +1,20 @@
1-
# global
1+
22
import copy
33
import gast
44
import re
55

6-
# local
7-
from ...configurations.base_transformer_config import (
6+
from ivy.transpiler.transformations.configurations.base_transformer_config import (
87
BaseTransformerConfig,
98
)
10-
from ...transformer import Transformer
11-
from ..base_transformer import (
9+
from ivy.transpiler.transformations.transformer import Transformer
10+
from ivy.transpiler.transformations.transformers.base_transformer import (
1211
BaseTransformer,
1312
)
14-
from ....utils.api_utils import (
13+
from ivy.transpiler.utils.api_utils import (
1514
TRANSLATED_OBJ_PREFIX,
1615
TRANSLATED_OBJ_SUFFIX,
1716
)
18-
from ....utils.ast_utils import ast_to_source_code, set_parents
17+
from ivy.transpiler.utils.ast_utils import ast_to_source_code, set_parents
1918

2019

2120
def clean_function_name(func_name):

ivy/transpiler/transformations/transformers/method_transformer/base_transformer.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,8 @@
1-
# local
2-
from ...configurations.base_transformer_config import (
1+
from ivy.transpiler.transformations.configurations.base_transformer_config import (
32
BaseTransformerConfig,
43
)
5-
from ...transformer import Transformer
6-
from ..base_transformer import (
4+
from ivy.transpiler.transformations.transformer import Transformer
5+
from ivy.transpiler.transformations.transformers.base_transformer import (
76
BaseTransformer,
87
)
98

ivy/transpiler/transformations/transformers/method_transformer/frontend_torch_method_transformer.py

Lines changed: 9 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1,33 +1,31 @@
1-
# global
21
import builtins
32
import gast
43

5-
# local
6-
from ...configurations.base_transformer_config import (
4+
from ivy.transpiler.transformations.configurations.base_transformer_config import (
75
BaseTransformerConfig,
86
)
9-
from ...transformer import Transformer
10-
from ....utils.conversion_utils import (
7+
from ivy.transpiler.transformations.transformer import Transformer
8+
from ivy.transpiler.utils.conversion_utils import (
119
BUILTIN_LIKELY_MODULE_NAMES,
1210
)
13-
from ..method_transformer.base_transformer import (
11+
from ivy.transpiler.transformations.transformers.method_transformer.base_transformer import (
1412
BaseMethodToFunctionConverter,
1513
)
16-
from ....utils.api_utils import (
14+
from ivy.transpiler.utils.api_utils import (
1715
is_property,
1816
is_builtin_method,
1917
is_method_of_class,
2018
get_frontend_class,
2119
is_frontend_api,
2220
is_ivy_api,
2321
)
24-
from ....utils.ast_utils import (
22+
from ivy.transpiler.utils.ast_utils import (
2523
ast_to_source_code,
2624
is_super_call_node,
2725
MODULE_TO_ALIAS,
2826
get_function_vars,
2927
)
30-
from ... import transformer_globals as glob
28+
from ivy.transpiler.transformations.transformer_globals import CONFLICTING_METHODS
3129

3230

3331
class FrontendTorchMethodToFunctionConverter(BaseMethodToFunctionConverter):
@@ -90,7 +88,7 @@ def visit_Attribute(self, node):
9088
node = new_call
9189

9290
# add the transformed method to the CONFLICTING_METHODS set
93-
glob.CONFLICTING_METHODS.add(right)
91+
CONFLICTING_METHODS.add(right)
9492

9593
return node
9694

@@ -163,7 +161,7 @@ def visit_Call(self, node):
163161
new_node = gast.Call(func=new_func, args=new_args, keywords=node.keywords)
164162

165163
# add the transformed method to the CONFLICTING_METHODS set
166-
glob.CONFLICTING_METHODS.add(right)
164+
CONFLICTING_METHODS.add(right)
167165

168166
"""
169167
Motivating example for using the @handle_methods decorator.

ivy/transpiler/transformations/transformers/method_transformer/ivy_method_transformer.py

Lines changed: 10 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,34 +1,32 @@
1-
# global
21
import gast
3-
import ivy
42
import re
53

6-
# local
7-
from ...configurations.base_transformer_config import (
4+
import ivy
5+
from ivy.transpiler.transformations.configurations.base_transformer_config import (
86
BaseTransformerConfig,
97
)
10-
from ...transformer import Transformer
11-
from ....utils.conversion_utils import (
8+
from ivy.transpiler.transformations.transformer import Transformer
9+
from ivy.transpiler.utils.conversion_utils import (
1210
BUILTIN_LIKELY_MODULE_NAMES,
1311
)
14-
from ..method_transformer.base_transformer import (
12+
from ivy.transpiler.transformations.transformers.method_transformer.base_transformer import (
1513
BaseMethodToFunctionConverter,
1614
)
17-
from ....utils.api_utils import (
15+
from ivy.transpiler.utils.api_utils import (
1816
is_property,
1917
is_builtin_method,
2018
is_method_of_class,
2119
is_frontend_api,
2220
is_ivy_api,
2321
get_function_from_modules,
2422
)
25-
from ....utils.ast_utils import (
23+
from ivy.transpiler.utils.ast_utils import (
2624
ast_to_source_code,
2725
is_super_call_node,
2826
MODULE_TO_ALIAS,
2927
get_function_vars,
3028
)
31-
from ... import transformer_globals as glob
29+
from ivy.transpiler.transformations.transformer_globals import CONFLICTING_METHODS
3230

3331

3432
class IvyMethodToFunctionConverter(BaseMethodToFunctionConverter):
@@ -107,7 +105,7 @@ def visit_Attribute(self, node):
107105
node = new_call
108106

109107
# add the transformed method to the CONFLICTING_METHODS set
110-
glob.CONFLICTING_METHODS.add(right)
108+
CONFLICTING_METHODS.add(right)
111109

112110
return node
113111

@@ -176,7 +174,7 @@ def visit_Call(self, node):
176174
new_node = gast.Call(func=new_func, args=new_args, keywords=node.keywords)
177175

178176
# add the transformed method to the CONFLICTING_METHODS set
179-
glob.CONFLICTING_METHODS.add(right)
177+
CONFLICTING_METHODS.add(right)
180178

181179
return new_node
182180

ivy/transpiler/transformations/transformers/method_transformer/native_torch_method_transformer.py

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,23 +1,21 @@
1-
# global
21
import gast
32
import importlib
43

5-
# local
6-
from ...configurations.base_transformer_config import (
4+
from ivy.transpiler.transformations.configurations.base_transformer_config import (
75
BaseTransformerConfig,
86
)
9-
from ...transformer import Transformer
10-
from ....utils.conversion_utils import (
7+
from ivy.transpiler.transformations.transformer import Transformer
8+
from ivy.transpiler.utils.conversion_utils import (
119
BUILTIN_LIKELY_MODULE_NAMES,
1210
)
13-
from ..method_transformer.base_transformer import (
11+
from ivy.transpiler.transformations.transformers.method_transformer.base_transformer import (
1412
BaseMethodToFunctionConverter,
1513
)
16-
from ....utils.api_utils import (
14+
from ivy.transpiler.utils.api_utils import (
1715
is_method_of_class,
1816
get_hf_class,
1917
)
20-
from ....utils.ast_utils import (
18+
from ivy.transpiler.utils.ast_utils import (
2119
ast_to_source_code,
2220
MODULE_TO_ALIAS,
2321
)

0 commit comments

Comments
 (0)