File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed
ivy_tests/test_transpiler/transformations Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff 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 ,
You can’t perform that action at this time.
0 commit comments