changes

georgia-tech-db · May 30, 2023 · 96d6901 · 96d6901
1 parent e28f609
commit 96d6901
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/eva/udfs/abstract/hf_abstract_udf.py b/eva/udfs/abstract/hf_abstract_udf.py
@@ -84,7 +84,9 @@ def output_formatter(self, outputs: Any):
             for row_output in outputs:
                 # account for the case where we have more than one prediction for an input
                 if isinstance(row_output, list):
-                    row_output = {k: [dic[k] for dic in row_output] for k in row_output[0]}
+                    row_output = {
+                        k: [dic[k] for dic in row_output] for k in row_output[0]
+                    }
                 result_list.append(row_output)
 
         result_df = pd.DataFrame(result_list)

diff --git a/test/integration_tests/test_huggingface_udfs.py b/test/integration_tests/test_huggingface_udfs.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2018-2022 EVA
+# Copyright 2018-2023 EVA
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.