Skip to content

Commit 026cee8

Browse files
committed
fix: transform_fn_w_list_comp test
1 parent e88116c commit 026cee8

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

ivy_tests/test_transpiler/transformations/test_transformations.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,8 @@ def test_transform_fn_w_list_comp(target):
2929
from ivy.transpiler.transformations.configurations.ivy_postprocessing_transformer_config import (
3030
IvyCodePostProcessorConfig,
3131
)
32-
from ivy.transpiler.transformations.transformers.postprocessing_transformer.ivy_postprocessing_transformer import (
33-
IvyCodePostProcessor,
32+
from ivy.transpiler.transformations.transformers.postprocessing_transformer.ivy_to_tf_postprocessing_transformer import (
33+
IvyToTFCodePostProcessor,
3434
)
3535

3636
# Set up the configurations container
@@ -43,7 +43,9 @@ def test_transform_fn_w_list_comp(target):
4343

4444
# Instantiate the transformer and transform the object
4545
configuration = IvyCodePostProcessorConfig()
46-
processor = IvyCodePostProcessor(
46+
47+
# TODO: generalize this for other target frameworks
48+
processor = IvyToTFCodePostProcessor(
4749
root,
4850
Transformer(object_like, container.translator_configurations[0]),
4951
configuration,

0 commit comments

Comments
 (0)