Fix

NVIDIA · Nov 28, 2023 · b6ad928 · b6ad928
1 parent 778fa00
commit b6ad928
Show file tree

Hide file tree

Showing 3 changed files with 595,884 additions and 2 deletions.
diff --git a/integration_tests/src/main/python/parquet_testing_test.py b/integration_tests/src/main/python/parquet_testing_test.py
@@ -17,7 +17,7 @@
 
 from asserts import assert_gpu_and_cpu_are_equal_collect, assert_gpu_and_cpu_error
 from conftest import get_std_input_path, is_parquet_testing_tests_forced, is_precommit_run, is_not_utc
-from data_gen import *
+from data_gen import copy_and_update, non_utc_allow
 from marks import allow_non_gpu
 from pathlib import Path
 import pytest

diff --git a/integration_tests/src/main/python/qa_nightly_select_test.py b/integration_tests/src/main/python/qa_nightly_select_test.py
@@ -25,7 +25,7 @@
 import pytest
 from spark_session import with_cpu_session, is_jvm_charset_utf8
 from marks import approximate_float, ignore_order, incompat, qarun, allow_non_gpu
-from data_gen import *
+from data_gen import copy_and_update, non_utc_allow
 
 def num_stringDf(spark):
     print("### CREATE DATAFRAME 1  ####")