moved execute query mapping

googleapis · Nov 12, 2024 · 39f12f1 · 39f12f1
1 parent d489ad3
commit 39f12f1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/google/cloud/bigtable/data/__init__.py b/google/cloud/bigtable/data/__init__.py
@@ -51,16 +51,12 @@
 )
 from google.cloud.bigtable.data._async._read_rows import _ReadRowsOperationAsync
 from google.cloud.bigtable.data._async._mutate_rows import _MutateRowsOperationAsync
-from google.cloud.bigtable.data.execute_query._async.execute_query_iterator import (
-    ExecuteQueryIteratorAsync,
-)
 
 from google.cloud.bigtable.data._cross_sync import CrossSync
 
 CrossSync.add_mapping("GapicClient", BigtableAsyncClient)
 CrossSync.add_mapping("_ReadRowsOperation", _ReadRowsOperationAsync)
 CrossSync.add_mapping("_MutateRowsOperation", _MutateRowsOperationAsync)
-CrossSync.add_mapping("ExecuteQueryIterator", ExecuteQueryIteratorAsync)
 CrossSync.add_mapping("MutationsBatcher", MutationsBatcherAsync)
 
 

diff --git a/google/cloud/bigtable/data/execute_query/__init__.py b/google/cloud/bigtable/data/execute_query/__init__.py
@@ -25,7 +25,9 @@
     QueryResultRow,
     Struct,
 )
+from google.cloud.bigtable.data._cross_sync import CrossSync
 
+CrossSync.add_mapping("ExecuteQueryIterator", ExecuteQueryIteratorAsync)
 
 __all__ = [
     "ExecuteQueryValueType",