[
https://issues.apache.org/jira/browse/BEAM-11587?focusedWorklogId=769395&page=com.atlassian.jira.plugin.system.issuetabpanels:worklog-tabpanel#worklog-769395
]
ASF GitHub Bot logged work on BEAM-11587:
-----------------------------------------
Author: ASF GitHub Bot
Created on: 12/May/22 02:39
Start Date: 12/May/22 02:39
Worklog Time Spent: 10m
Work Description: svetakvsundhar commented on code in PR #17159:
URL: https://github.com/apache/beam/pull/17159#discussion_r870899342
##########
sdks/python/apache_beam/io/gcp/bigquery.py:
##########
@@ -2525,6 +2526,12 @@ def _get_pipeline_details(unused_elm):
**self._kwargs))
| _PassThroughThenCleanupTempDatasets(project_to_cleanup_pcoll))
+ def get_pcoll_from_schema(table_schema):
+ pcoll_val = apache_beam.io.gcp.bigquery_schema_tools.\
+ produce_pcoll_with_schema(table_schema)
+ return beam.Map(lambda values: pcoll_val(**values)).with_output_types(
Review Comment:
Ah ok I think I've found the RC:
https://cloud.google.com/dataflow/docs/resources/faq#how_do_i_handle_nameerrors
Looks like the pickling is happen in my global namespace, and not carrying
over to the Dataflow worker. A fix such as passing in ```--save_main_session```
should do the trick. See https://issues.apache.org/jira/browse/BEAM-6158 for a
similar error that is tangentially related.
Issue Time Tracking
-------------------
Worklog Id: (was: 769395)
Time Spent: 8.5h (was: 8h 20m)
> Support pd.read_gbq and DataFrame.to_gbq
> ----------------------------------------
>
> Key: BEAM-11587
> URL: https://issues.apache.org/jira/browse/BEAM-11587
> Project: Beam
> Issue Type: New Feature
> Components: dsl-dataframe, io-py-gcp, sdk-py-core
> Reporter: Brian Hulette
> Assignee: Svetak Vihaan Sundhar
> Priority: P3
> Labels: dataframe-api
> Time Spent: 8.5h
> Remaining Estimate: 0h
>
> We should support
> [read_gbq|https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.read_gbq.html]
> andÂ
> [to_gbq|https://pandas.pydata.org/docs/reference/api/pandas.DataFrame.to_gbq.html]
> in the DataFrame API when gcp extras are installed.
--
This message was sent by Atlassian Jira
(v8.20.7#820007)
