From 026cee8cbd281c68225f168dfd7ab95e63611774 Mon Sep 17 00:00:00 2001
From: Sam Armstrong <88863522+Sam-Armstrong@users.noreply.github.com>
Date: Thu, 16 Jan 2025 19:24:01 +0000
Subject: [PATCH] fix: transform_fn_w_list_comp test

---
 .../transformations/test_transformations.py               | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/ivy_tests/test_transpiler/transformations/test_transformations.py b/ivy_tests/test_transpiler/transformations/test_transformations.py
index 1c9205fcf8538..439dfec3e3aa4 100644
--- 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,