Skip to content

Commit

Permalink
formatting
Browse files Browse the repository at this point in the history
Signed-off-by: niklasvm <[email protected]>
  • Loading branch information
niklasvm committed Aug 11, 2022
1 parent 4754100 commit 2ad0093
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -7,11 +7,11 @@
import pandas
import pandas as pd
import pyarrow
import pyarrow.parquet as pq
import pyspark
from pydantic import StrictStr
from pyspark import SparkConf
from pyspark.sql import SparkSession
import pyarrow.parquet as pq
from pytz import utc

from feast import FeatureView, OnDemandFeatureView
Expand Down Expand Up @@ -272,7 +272,7 @@ def _to_arrow_internal(self) -> pyarrow.Table:

# write to temp parquet and then load it as pyarrow table from disk
with tempfile.TemporaryDirectory() as temp_dir:
self.to_spark_df().write.parquet(temp_dir,mode="overwrite")
self.to_spark_df().write.parquet(temp_dir, mode="overwrite")
return pq.read_table(temp_dir)

def persist(self, storage: SavedDatasetStorage):
Expand Down

0 comments on commit 2ad0093

Please sign in to comment.