fix formatting

Author: ohmayr

gapic/cli/generate.py

@@ -25,6 +25,7 @@
 from gapic.utils import Options
 from gapic.utils.cache import generation_cache_context
 
+
 @click.command()
 @click.option(
     "--request",

gapic/schema/metadata.py

@@ -361,9 +361,7 @@ def resolve(self, selector: str) -> str:
         return selector
 
     @cached_proto_context
-    def with_context(
-        self, *, collisions: Set[str]
-    ) -> "Address":
+    def with_context(self, *, collisions: Set[str]) -> "Address":
         """Return a derivative of this address with the provided context.
 
         This method is used to address naming collisions. The returned
@@ -403,9 +401,7 @@ def doc(self):
         return ""
 
     @cached_proto_context
-    def with_context(
-        self, *, collisions: Set[str]
-    ) -> "Metadata":
+    def with_context(self, *, collisions: Set[str]) -> "Metadata":
         """Return a derivative of this metadata with the provided context.
 
         This method is used to address naming collisions. The returned
@@ -415,9 +411,7 @@ def with_context(
         return (
             dataclasses.replace(
                 self,
-                address=self.address.with_context(
-                    collisions=collisions
-                ),
+                address=self.address.with_context(collisions=collisions),
             )
             if collisions and collisions != self.address.collisions
             else self

gapic/schema/wrappers.py

@@ -437,13 +437,7 @@ def with_context(
                 if self.message
                 else None
             ),
-            enum=(
-                self.enum.with_context(
-                    collisions=collisions
-                )
-                if self.enum
-                else None
-            ),
+            enum=(self.enum.with_context(collisions=collisions) if self.enum else None),
             meta=self.meta.with_context(
                 collisions=collisions,
             ),
@@ -862,9 +856,7 @@ def with_context(
                 )
                 for k, v in self.nested_messages.items()
             },
-            meta=self.meta.with_context(
-                collisions=collisions
-            ),
+            meta=self.meta.with_context(collisions=collisions),
         )
 
     def add_to_address_allowlist(
@@ -968,9 +960,7 @@ def with_context(
         return (
             dataclasses.replace(
                 self,
-                meta=self.meta.with_context(
-                    collisions=collisions
-                ),
+                meta=self.meta.with_context(collisions=collisions),
             )
             if collisions
             else self

gapic/utils/cache.py

@@ -47,6 +47,7 @@ def inner(self):
 
     return property(inner)
 
+
 # 1. The ContextVar (Default is None)
 # This replaces threading.local with the async-safe standard.
 generation_cache: ContextVar[Optional[Dict[Any, Any]]] = ContextVar(
@@ -56,10 +57,11 @@ def inner(self):
 # Optimization: Bind .get for speed
 _get_cache = generation_cache.get
 
+
 @contextlib.contextmanager
 def generation_cache_context():
     """Context manager to explicitly manage the cache lifecycle.
-    
+
     Usage:
         with generation_cache_context():
             # Cache is active (fast)
@@ -74,6 +76,7 @@ def generation_cache_context():
         # This allows the dictionary to be garbage collected.
         generation_cache.reset(token)
 
+
 def cached_proto_context(func):
     """Decorator to memoize with_context calls based on self and collisions."""
 
@@ -83,7 +86,7 @@ def wrapper(self, *, collisions, **kwargs):
 
         context_cache = _get_cache()
         if context_cache is None:
-             raise RuntimeError(
+            raise RuntimeError(
                 f"Cache MISSING! {func.__name__} called on {type(self).__name__} outside of context manager."
             )
 
@@ -97,9 +100,7 @@ def wrapper(self, *, collisions, **kwargs):
 
         # 4. Execute the actual function
         # We ensure context_cache is passed down to the recursive calls
-        result = func(
-            self, collisions=collisions, **kwargs
-        )
+        result = func(self, collisions=collisions, **kwargs)
 
         # 5. Update Cache
         context_cache[key] = result