fix: transform_fn_w_list_comp test

ivy-llc · Jan 16, 2025 · 026cee8 · 026cee8
1 parent e88116c
commit 026cee8
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/ivy_tests/test_transpiler/transformations/test_transformations.py b/ivy_tests/test_transpiler/transformations/test_transformations.py
@@ -29,8 +29,8 @@ def test_transform_fn_w_list_comp(target):
     from ivy.transpiler.transformations.configurations.ivy_postprocessing_transformer_config import (
         IvyCodePostProcessorConfig,
     )
-    from ivy.transpiler.transformations.transformers.postprocessing_transformer.ivy_postprocessing_transformer import (
-        IvyCodePostProcessor,
+    from ivy.transpiler.transformations.transformers.postprocessing_transformer.ivy_to_tf_postprocessing_transformer import (
+        IvyToTFCodePostProcessor,
     )
 
     # Set up the configurations container
@@ -43,7 +43,9 @@ def test_transform_fn_w_list_comp(target):
 
     # Instantiate the transformer and transform the object
     configuration = IvyCodePostProcessorConfig()
-    processor = IvyCodePostProcessor(
+
+    # TODO: generalize this for other target frameworks
+    processor = IvyToTFCodePostProcessor(
         root,
         Transformer(object_like, container.translator_configurations[0]),
         configuration,