return type

apache · Sep 27, 2024 · 4e0ad87 · 4e0ad87
1 parent 67c8349
commit 4e0ad87
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/sdks/python/apache_beam/coders/coder_impl.py b/sdks/python/apache_beam/coders/coder_impl.py
@@ -74,6 +74,7 @@
 if TYPE_CHECKING:
   import proto
   from apache_beam.transforms import userstate
+  from apache_beam.transforms.window import BoundedWindow
   from apache_beam.transforms.window import IntervalWindow
 
 try:
@@ -806,6 +807,7 @@ def estimate_size(self, unused_value, nested=False):
 
 if not TYPE_CHECKING:
   IntervalWindow = None
+  BoundedWindow = None
 
 
 class IntervalWindowCoderImpl(StreamCoderImpl):
@@ -834,7 +836,7 @@ def encode_to_stream(self, value, out, nested):
     out.write_var_int64(span_millis)
 
   def decode_from_stream(self, in_, nested):
-    # type: (create_InputStream, bool) -> IntervalWindow
+    # type: (create_InputStream, bool) -> BoundedWindow
     if not TYPE_CHECKING:
       global IntervalWindow  # pylint: disable=global-variable-not-assigned
       if IntervalWindow is None: