Skip to content
This repository was archived by the owner on Apr 1, 2026. It is now read-only.

Commit d989eb2

Browse files
fix arg inconsistencies
1 parent 8b5bca7 commit d989eb2

File tree

2 files changed

+5
-7
lines changed

2 files changed

+5
-7
lines changed

bigframes/pandas/io/api.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -620,14 +620,12 @@ def read_orc(
620620
def read_avro(
621621
path: str | IO["bytes"],
622622
*,
623-
engine: str = "bigquery",
624-
write_engine: constants.WriteEngineType = "default",
623+
engine: str = "auto",
625624
) -> bigframes.dataframe.DataFrame:
626625
return global_session.with_default_session(
627626
bigframes.session.Session.read_avro,
628627
path,
629628
engine=engine,
630-
write_engine=write_engine,
631629
)
632630

633631

bigframes/session/__init__.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1424,23 +1424,23 @@ def read_avro(
14241424
self,
14251425
path: str | IO["bytes"],
14261426
*,
1427-
engine: str = "default",
1427+
engine: str = "auto",
14281428
) -> dataframe.DataFrame:
14291429
"""Load an Avro file to a BigQuery DataFrames DataFrame.
14301430
14311431
Args:
14321432
path (str or IO):
14331433
The path or buffer to the Avro file. Can be a local path or Google Cloud Storage URI.
1434-
engine (str, default "default"):
1434+
engine (str, default "auto"):
14351435
The engine used to read the file. Only `bigquery` is supported for Avro.
14361436
14371437
Returns:
14381438
bigframes.dataframe.DataFrame:
14391439
A new DataFrame representing the data from the Avro file.
14401440
"""
1441-
if engine not in ("default", "bigquery"):
1441+
if engine not in ("auto", "bigquery"):
14421442
raise ValueError(
1443-
f"Unsupported engine: {repr(engine)}. Supported values: 'default', 'bigquery'."
1443+
f"Unsupported engine: {repr(engine)}. Supported values: 'auto', 'bigquery'."
14441444
)
14451445

14461446
job_config = bigquery.LoadJobConfig()

0 commit comments

Comments
 (0)