Ejemplo n.º 1
0
 def do_it_all(spark):
     df = df_fun(spark)
     df.createOrReplaceTempView(table_name)
     if debug:
         return data_gen.debug_df(spark.sql(sql))
     else:
         return spark.sql(sql)
Ejemplo n.º 2
0
    def do_it_all(spark):
        df = df_fun(spark)
        df.createOrReplaceTempView(table_name)
        # we hold off on setting the validate execs until after creating the temp view

        spark.conf.set('spark.rapids.sql.test.validateExecsInGpuPlan', ','.join(validate_execs_in_gpu_plan))
        if debug:
            return data_gen.debug_df(spark.sql(sql))
        else:
            return spark.sql(sql)