Adjust sample in client docstring

googleapis · Sep 22, 2020 · cb753e5 · cb753e5
1 parent 373cb19
commit cb753e5
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/google/cloud/bigquery_storage_v1/client.py b/google/cloud/bigquery_storage_v1/client.py
@@ -76,9 +76,11 @@ def read_rows(
             >>>
             >>> requested_session = bigquery_storage_v1.types.ReadSession(
             ...     table=table,
-            ...     data_format=bigquery_storage_v1.enums.DataFormat.AVRO,
+            ...     data_format=bigquery_storage_v1.types.DataFormat.AVRO,
+            ... )
+            >>> session = client.create_read_session(
+            ...     parent=parent, read_session=requested_session
             ... )
-            >>> session = client.create_read_session(parent, requested_session)
             >>>
             >>> stream = session.streams[0],  # TODO: Also read any other streams.
             >>> read_rows_stream = client.read_rows(stream.name)