fix: wrong genereated type hints

googleapis · Jan 29, 2021 · 86e2c43 · 86e2c43
1 parent 48253d3
commit 86e2c43
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gapic/templates/%namespace/%name_%version/%sub/services/%service/pagers.py.j2 b/gapic/templates/%namespace/%name_%version/%sub/services/%service/pagers.py.j2
@@ -69,11 +69,11 @@ class {{ method.name }}Pager:
             yield self._response
 
     {% if method.paged_result_field.map %}
-    def __iter__(self) -> Iterable[Tuple[str, {{ method.paged_result_field.ident | replace('Sequence[', '') | replace(']', '') }}]]:
+    def __iter__(self) -> Iterable[Tuple[str, {{ method.paged_result_field.type.fields.get('value').ident }}]]:
         for page in self.pages:
             yield from page.{{ method.paged_result_field.name}}.items()
 
-    def get(self, key: str) -> {{ method.paged_result_field.ident | replace('Sequence', 'Optional') }}:
+    def get(self, key: str) -> Optional[{{ method.paged_result_field.type.fields.get('value').ident }}]:
         return self._response.items.get(key)
     {% else %}
     def __iter__(self) -> {{ method.paged_result_field.ident | replace('Sequence', 'Iterable') }}: