fix arguments mode

tricktx · tricktx · commit 9374e9d7dedb · 2026-03-08T23:02:59.000-03:00
diff --git a/python-package/basedosdados/upload/table.py b/python-package/basedosdados/upload/table.py
@@ -150,10 +150,10 @@ def _load_schema_from_bq(
         """Load schema from table config
 
         Args:
-            mode: Which dataset to create [`prod`|`staging`].
+            project_gcp: Which dataset to create [`prod`|`staging`].
 
         """
-        table_columns = self._get_columns_from_bq()
+        table_columns = self._get_columns_from_bq(project_gcp=project_gcp)
         columns = table_columns.get("partition_columns") + table_columns.get(
             "columns"
         )
diff --git a/python-package/tests/test_table.py b/python-package/tests/test_table.py
@@ -211,7 +211,6 @@ def test_get_columns_from_data():
     out2 = table._get_columns_from_data(
         data_sample_path="tests/sample_data/table",
         source_format="csv",
-        mode="staging",
     )
 
     assert isinstance(out, dict)

Original file line number	Diff line number	Diff line change
`@@ -211,7 +211,6 @@ def test_get_columns_from_data():`
`211`	`211`	`out2 = table._get_columns_from_data(`
`212`	`212`	`data_sample_path="tests/sample_data/table",`
`213`	`213`	`source_format="csv",`
`214`		`- mode="staging",`
`215`	`214`	`)`
`216`	`215`
`217`	`216`	`assert isinstance(out, dict)`