Update sdk/python/feast/infra/offline_stores/bigquery.py

Co-authored-by: Daniel Salvador <[email protected]>
vinted · Nov 7, 2023 · 7374e78 · 7374e78
1 parent 70867d8
commit 7374e78
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdk/python/feast/infra/offline_stores/bigquery.py b/sdk/python/feast/infra/offline_stores/bigquery.py
@@ -574,7 +574,7 @@ def to_remote_storage(self) -> List[str]:
         table = self.to_bigquery()
 
         parquet_file_size_mb = os.getenv("BQ_EXPORT_PARQUET_FILE_SIZE_MB")
-        if parquet_file_size_mb is not None:
+        if parquet_file_size_mb is not None and parquet_file_size_mb.isdigit():
             table_size_in_mb = self.client.get_table(table).num_bytes / 1024 / 1024
             number_of_files = max(1, table_size_in_mb // int(parquet_file_size_mb))
             destination_uris = [