import pyspark.sql.functions as F


class MockClass:

    def execute(self, df):
        df = df.withColumn('y', F.lit('6'))
        return df

    def udf_action(self, value):
        print(6)

a66b26aeb63d3b7015138c3d94d1b82101cd417b
<mask token>

<mask token>


class MockClass:

    def execute(self, df):
        df = df.withColumn('y', F.lit('6'))
        return df
    <mask token>

<mask token>


class MockClass:

    def execute(self, df):
        df = df.withColumn('y', F.lit('6'))
        return df

    def udf_action(self, value):
        print(6)

import pyspark.sql.functions as F


class MockClass:

    def execute(self, df):
        df = df.withColumn('y', F.lit('6'))
        return df

    def udf_action(self, value):
        print(6)
