Only use arg when pyarrow>=4.0.0

judahrand · judahrand · commit 4e46f8317447 · 2021-09-22T17:41:52.000+01:00
diff --git a/google/cloud/bigquery/_pandas_helpers.py b/google/cloud/bigquery/_pandas_helpers.py
@@ -570,18 +570,23 @@ def dataframe_to_parquet(
             compliant Parquet nested type (lists). Defaults to ``True``.
             https://github.com/apache/parquet-format/blob/master/LogicalTypes.md#nested-types
             https://arrow.apache.org/docs/python/generated/pyarrow.parquet.write_table.html#pyarrow-parquet-write-table
+
+            This argument is ignored for ``pyarrow`` versions earlier than ``4.0.0``.
     """
     pyarrow = _helpers.PYARROW_VERSIONS.try_import(raise_if_error=True)
 
     import pyarrow.parquet
 
+    kwargs = (
+        {"use_compliant_nested_type": parquet_use_compliant_nested_type}
+        if _helpers._PYARROW_VERSION.major >= 4
+        else {}
+    )
+
     bq_schema = schema._to_schema_fields(bq_schema)
     arrow_table = dataframe_to_arrow(dataframe, bq_schema)
     pyarrow.parquet.write_table(
-        arrow_table,
-        filepath,
-        compression=parquet_compression,
-        use_compliant_nested_type=parquet_use_compliant_nested_type,
+        arrow_table, filepath, compression=parquet_compression, **kwargs,
     )
 
 
diff --git a/google/cloud/bigquery/client.py b/google/cloud/bigquery/client.py
@@ -2547,6 +2547,8 @@ def load_table_from_dataframe(
                 ``DataFrame.to_parquet()`` method.
                 https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.to_parquet.html#pandas.DataFrame.to_parquet
 
+                This argument is ignored for ``pyarrow`` versions before ``4.0.0``.
+
                 This argument is only present to allow for backwards compatibility with
                 tables created using an old version of this method.
             timeout (Optional[float]):
@@ -2679,7 +2681,11 @@ def load_table_from_dataframe(
                         tmppath,
                         engine="pyarrow",
                         compression=parquet_compression,
-                        use_compliant_nested_type=parquet_use_compliant_nested_type,
+                        **{
+                            "use_compliant_nested_type": parquet_use_compliant_nested_type
+                        }
+                        if _PYARROW_VERSION.major >= 4
+                        else {},
                     )
 
             else:
