Ensure that number_of_files is int (#15)

vinted · Nov 8, 2023 · 8452624 · 8452624
1 parent 6df40b0
commit 8452624
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
@@ -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)