From 7f9eaec50b522d39ef8ffd9acd19b3ec8f008bdf Mon Sep 17 00:00:00 2001
From: niehaus59 <41398747+niehaus59@users.noreply.github.com>
Date: Sat, 25 Jun 2022 16:04:52 -0700
Subject: [PATCH] fix: remove debugging code (#1546)

* remove unused imports

* batch prompts

* delete debugging code
---
 .../azure/synapse/ml/cognitive/split1/OpenAISuite.scala     | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/cognitive/src/test/scala/com/microsoft/azure/synapse/ml/cognitive/split1/OpenAISuite.scala b/cognitive/src/test/scala/com/microsoft/azure/synapse/ml/cognitive/split1/OpenAISuite.scala
index ffd906a86d..034a96d438 100644
--- a/cognitive/src/test/scala/com/microsoft/azure/synapse/ml/cognitive/split1/OpenAISuite.scala
+++ b/cognitive/src/test/scala/com/microsoft/azure/synapse/ml/cognitive/split1/OpenAISuite.scala
@@ -82,13 +82,9 @@ class OpenAICompletionSuite extends TransformerFuzzing[OpenAICompletion] with Op
 
   def testCompletion(completion: OpenAICompletion, df: DataFrame, requiredLength: Int = 10): Unit = {
     val fromRow = CompletionResponse.makeFromRowConverter
-    val t = completion.transform(df)
-    t.collect().map(r =>
+    completion.transform(df).collect().map(r =>
       fromRow(r.getAs[Row]("out")).choices.map(c =>
         assert(c.text.length > requiredLength)))
-    /*completion.transform(df).collect().map(r =>
-      fromRow(r.getAs[Row]("out")).choices.map(c =>
-        assert(c.text.length > requiredLength)))*/
   }
 
   def newCompletion(): OpenAICompletion = {