def count(self, df: DataFrame) -> int:
if df.is_bounded:
return df.count()
else:
return sum(1 for _ in df.as_array_iterable())
def t3(df1: DataFrame, df2: DataFrame, a) -> DataFrame:
value = df1.count() + df2.count() + a
return ArrayDataFrame([[value]], "a:int")
def _persist_and_count(df: DataFrame) -> int:
df = self.persist(df)
return df.count()
def t3(df1: DataFrame, df2: DataFrame, a, b) -> None:
b.value = df1.count() + df2.count() + a
