make function private

niklasvm · niklasvm · commit f8e70ea784ce · 2022-09-05T11:35:20.000+02:00
Signed-off-by: niklasvm &lt;niklasvm@gmail.com&gt;
diff --git a/sdk/python/feast/infra/materialization/contrib/spark/spark_materialization_engine.py b/sdk/python/feast/infra/materialization/contrib/spark/spark_materialization_engine.py
@@ -178,7 +178,7 @@ def _materialize_one(
             # split data into batches
             spark_df = offline_job.to_spark_df()
             batch_size = self.repo_config.batch_engine.batch_size
-            batched_spark_df, batch_column_alias = add_batch_column(
+            batched_spark_df, batch_column_alias = _add_batch_column(
                 spark_df,
                 join_key_columns=join_key_columns,
                 timestamp_field=timestamp_field,
@@ -209,7 +209,7 @@ def _materialize_one(
             )
 
 
-def add_batch_column(
+def _add_batch_column(
     spark_df: DataFrame, join_key_columns, timestamp_field, batch_size
 ):
     """
