From e2ba052082d41874eef9c929fa318e8f384b2fb5 Mon Sep 17 00:00:00 2001 From: Martynas Jurkus Date: Wed, 8 Nov 2023 14:57:45 +0200 Subject: [PATCH] Ensure that number_of_files is int --- sdk/python/feast/infra/offline_stores/bigquery.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdk/python/feast/infra/offline_stores/bigquery.py b/sdk/python/feast/infra/offline_stores/bigquery.py index 57aaa192dd..39c13202b0 100644 --- a/sdk/python/feast/infra/offline_stores/bigquery.py +++ b/sdk/python/feast/infra/offline_stores/bigquery.py @@ -581,7 +581,7 @@ def to_remote_storage(self) -> List[str]: table_size_in_mb = self.client.get_table(table).num_bytes / 1024 / 1024 number_of_files = max( 1, - table_size_in_mb // self.config.offline_store.gcs_staging_file_size_mb, + int(table_size_in_mb // self.config.offline_store.gcs_staging_file_size_mb), ) destination_uris = [ f"{self._gcs_path}/{n:0>12}.parquet" for n in range(number_of_files)