Skip to content

Commit 87bc81d

Browse files
committed
Fix value names
1 parent e6639e7 commit 87bc81d

1 file changed

Lines changed: 9 additions & 9 deletions

File tree

databricks/koalas/indexes.py

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

15851585
if isinstance(value, (list, tuple, Index, Series)):
15861586
if isinstance(value, (list, tuple)):
1587-
pandas_replace_value = pd.Series(value)
1587+
pandas_value = pd.Series(value)
15881588
elif isinstance(value, (Index, Series)):
1589-
pandas_replace_value = value.to_pandas()
1589+
pandas_value = value.to_pandas()
15901590

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

1595-
replace_return_type = as_spark_type(pandas_replace_value.dtype.type)
1595+
replace_return_type = as_spark_type(pandas_value.dtype.type)
15961596

15971597
@pandas_udf(returnType=replace_return_type if replace_return_type else StringType())
15981598
def replace_pandas_udf(sequence):
1599-
return pandas_replace_value[sequence]
1599+
return pandas_value[sequence]
16001600

16011601
sdf = sdf.withColumn(replace_col, replace_pandas_udf(dist_sequence_col_name))
16021602
else:
16031603
sdf = sdf.withColumn(replace_col, F.lit(value))
16041604

16051605
if isinstance(mask, (list, tuple, Index, Series)):
16061606
if isinstance(mask, (list, tuple)):
1607-
pandas_replace_mask = pd.Series(mask)
1607+
pandas_mask = pd.Series(mask)
16081608
elif isinstance(mask, (Index, Series)):
1609-
pandas_replace_mask = mask.to_pandas()
1609+
pandas_mask = mask.to_pandas()
16101610

1611-
if self.size != pandas_replace_mask.size:
1611+
if self.size != pandas_mask.size:
16121612
raise ValueError("mask and data must be the same size")
16131613

16141614
@pandas_udf(returnType=BooleanType())
16151615
def masking_pandas_udf(sequence):
1616-
return pandas_replace_mask[sequence]
1616+
return pandas_mask[sequence]
16171617

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

0 commit comments

Comments
 (0)