Skip to content

Commit 34c6963

Browse files
committed
ruff fix
1 parent 714981e commit 34c6963

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

python/hsfs/engine/spark.py

+6-6
Original file line numberDiff line numberDiff line change
@@ -1329,9 +1329,9 @@ def save_empty_dataframe(self, feature_group):
13291329

13301330
dataframe = self._spark_session.read.format("hudi").load(location)
13311331

1332-
for feature in feature_group.features:
1333-
if feature.name not in dataframe.columns:
1334-
dataframe = dataframe.withColumn(feature.name, lit(None).cast(feature.type))
1332+
for _feature in feature_group.features:
1333+
if _feature.name not in dataframe.columns:
1334+
dataframe = dataframe.withColumn(_feature.name, lit(None).cast(_feature.type))
13351335

13361336
self.save_dataframe(
13371337
feature_group,
@@ -1348,9 +1348,9 @@ def add_cols_to_delta_table(self, feature_group):
13481348

13491349
dataframe = self._spark_session.read.format("delta").load(location)
13501350

1351-
for feature in feature_group.features:
1352-
if feature.name not in dataframe.columns:
1353-
dataframe = dataframe.withColumn(feature.name, lit(None).cast(feature.type))
1351+
for _feature in feature_group.features:
1352+
if _feature.name not in dataframe.columns:
1353+
dataframe = dataframe.withColumn(_feature.name, lit(None).cast(_feature.type))
13541354

13551355
dataframe.limit(0).write.format("delta").mode(
13561356
"append"

0 commit comments

Comments
 (0)