fix: remove debugging code (#1546)

* remove unused imports * batch prompts * delete debugging code
microsoft · Jun 25, 2022 · 7f9eaec · 7f9eaec
1 parent 137577d
commit 7f9eaec
Showing 1 changed file with 1 addition and 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
@@ -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 = {