Skip to content

Commit 097cc3a

Browse files
committed
Fix value names
1 parent 9bd21b1 commit 097cc3a

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

databricks/koalas/indexes.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1590,36 +1590,36 @@ def putmask(self, mask, value):
15901590

15911591
if isinstance(value, (list, tuple, Index, Series)):
15921592
if isinstance(value, (list, tuple)):
1593-
pandas_replace_value = pd.Series(value)
1593+
pandas_value = pd.Series(value)
15941594
elif isinstance(value, (Index, Series)):
1595-
pandas_replace_value = value.to_pandas()
1595+
pandas_value = value.to_pandas()
15961596

1597-
if self.size != pandas_replace_value.size:
1597+
if self.size != pandas_value.size:
15981598
# TODO: We can't support different size of value for now.
15991599
raise ValueError("value and data must be the same size")
16001600

1601-
replace_return_type = as_spark_type(pandas_replace_value.dtype.type)
1601+
replace_return_type = as_spark_type(pandas_value.dtype.type)
16021602

16031603
@pandas_udf(returnType=replace_return_type if replace_return_type else StringType())
16041604
def replace_pandas_udf(sequence):
1605-
return pandas_replace_value[sequence]
1605+
return pandas_value[sequence]
16061606

16071607
sdf = sdf.withColumn(replace_col, replace_pandas_udf(dist_sequence_col_name))
16081608
else:
16091609
sdf = sdf.withColumn(replace_col, F.lit(value))
16101610

16111611
if isinstance(mask, (list, tuple, Index, Series)):
16121612
if isinstance(mask, (list, tuple)):
1613-
pandas_replace_mask = pd.Series(mask)
1613+
pandas_mask = pd.Series(mask)
16141614
elif isinstance(mask, (Index, Series)):
1615-
pandas_replace_mask = mask.to_pandas()
1615+
pandas_mask = mask.to_pandas()
16161616

1617-
if self.size != pandas_replace_mask.size:
1617+
if self.size != pandas_mask.size:
16181618
raise ValueError("mask and data must be the same size")
16191619

16201620
@pandas_udf(returnType=BooleanType())
16211621
def masking_pandas_udf(sequence):
1622-
return pandas_replace_mask[sequence]
1622+
return pandas_mask[sequence]
16231623

16241624
sdf = sdf.withColumn(masking_col, masking_pandas_udf(dist_sequence_col_name))
16251625
elif not isinstance(mask, list) and not isinstance(mask, tuple):

0 commit comments

Comments
 (0)