chore: Protobuf generated files

.github/workflows/lint.yml

@@ -14,7 +14,7 @@ jobs:
       - uses: actions/checkout@v3
       - uses: psf/black@stable
         with:
-          options: "--check --verbose"
+          options: "--check --verbose --exclude=pynumaflow/proto/"
           version: "~= 23.3"
 
   ruff:

Makefile

@@ -7,7 +7,7 @@ clean:
 	@find . -type f -name "*.py[co]" -exec rm -rf {} +
 
 format: clean
-	poetry run black pynumaflow/ tests/ examples/
+	poetry run black --exclude=pynumaflow/proto/ pynumaflow/ tests/ examples/
 
 
 lint: format

pynumaflow/proto/accumulator/accumulator_pb2.pyi

@@ -1,22 +1,18 @@
+import datetime
+
 from google.protobuf import empty_pb2 as _empty_pb2
 from google.protobuf import timestamp_pb2 as _timestamp_pb2
 from google.protobuf.internal import containers as _containers
 from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import message as _message
-from typing import (
-    ClassVar as _ClassVar,
-    Iterable as _Iterable,
-    Mapping as _Mapping,
-    Optional as _Optional,
-    Union as _Union,
-)
+from collections.abc import Iterable as _Iterable, Mapping as _Mapping
+from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
 
 DESCRIPTOR: _descriptor.FileDescriptor
 
 class Payload(_message.Message):
     __slots__ = ("keys", "value", "event_time", "watermark", "id", "headers")
-
     class HeadersEntry(_message.Message):
         __slots__ = ("key", "value")
         KEY_FIELD_NUMBER: _ClassVar[int]
@@ -36,22 +32,12 @@ class Payload(_message.Message):
     watermark: _timestamp_pb2.Timestamp
     id: str
     headers: _containers.ScalarMap[str, str]
-    def __init__(
-        self,
-        keys: _Optional[_Iterable[str]] = ...,
-        value: _Optional[bytes] = ...,
-        event_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
-        watermark: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
-        id: _Optional[str] = ...,
-        headers: _Optional[_Mapping[str, str]] = ...,
-    ) -> None: ...
+    def __init__(self, keys: _Optional[_Iterable[str]] = ..., value: _Optional[bytes] = ..., event_time: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., watermark: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., id: _Optional[str] = ..., headers: _Optional[_Mapping[str, str]] = ...) -> None: ...
 
 class AccumulatorRequest(_message.Message):
     __slots__ = ("payload", "operation")
-
     class WindowOperation(_message.Message):
         __slots__ = ("event", "keyedWindow")
-
         class Event(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
             __slots__ = ()
             OPEN: _ClassVar[AccumulatorRequest.WindowOperation.Event]
@@ -64,20 +50,12 @@ class AccumulatorRequest(_message.Message):
         KEYEDWINDOW_FIELD_NUMBER: _ClassVar[int]
         event: AccumulatorRequest.WindowOperation.Event
         keyedWindow: KeyedWindow
-        def __init__(
-            self,
-            event: _Optional[_Union[AccumulatorRequest.WindowOperation.Event, str]] = ...,
-            keyedWindow: _Optional[_Union[KeyedWindow, _Mapping]] = ...,
-        ) -> None: ...
+        def __init__(self, event: _Optional[_Union[AccumulatorRequest.WindowOperation.Event, str]] = ..., keyedWindow: _Optional[_Union[KeyedWindow, _Mapping]] = ...) -> None: ...
     PAYLOAD_FIELD_NUMBER: _ClassVar[int]
     OPERATION_FIELD_NUMBER: _ClassVar[int]
     payload: Payload
     operation: AccumulatorRequest.WindowOperation
-    def __init__(
-        self,
-        payload: _Optional[_Union[Payload, _Mapping]] = ...,
-        operation: _Optional[_Union[AccumulatorRequest.WindowOperation, _Mapping]] = ...,
-    ) -> None: ...
+    def __init__(self, payload: _Optional[_Union[Payload, _Mapping]] = ..., operation: _Optional[_Union[AccumulatorRequest.WindowOperation, _Mapping]] = ...) -> None: ...
 
 class KeyedWindow(_message.Message):
     __slots__ = ("start", "end", "slot", "keys")
@@ -89,13 +67,7 @@ class KeyedWindow(_message.Message):
     end: _timestamp_pb2.Timestamp
     slot: str
     keys: _containers.RepeatedScalarFieldContainer[str]
-    def __init__(
-        self,
-        start: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
-        end: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
-        slot: _Optional[str] = ...,
-        keys: _Optional[_Iterable[str]] = ...,
-    ) -> None: ...
+    def __init__(self, start: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., end: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., slot: _Optional[str] = ..., keys: _Optional[_Iterable[str]] = ...) -> None: ...
 
 class AccumulatorResponse(_message.Message):
     __slots__ = ("payload", "window", "tags", "EOF")
@@ -107,13 +79,7 @@ class AccumulatorResponse(_message.Message):
     window: KeyedWindow
     tags: _containers.RepeatedScalarFieldContainer[str]
     EOF: bool
-    def __init__(
-        self,
-        payload: _Optional[_Union[Payload, _Mapping]] = ...,
-        window: _Optional[_Union[KeyedWindow, _Mapping]] = ...,
-        tags: _Optional[_Iterable[str]] = ...,
-        EOF: bool = ...,
-    ) -> None: ...
+    def __init__(self, payload: _Optional[_Union[Payload, _Mapping]] = ..., window: _Optional[_Union[KeyedWindow, _Mapping]] = ..., tags: _Optional[_Iterable[str]] = ..., EOF: bool = ...) -> None: ...
 
 class ReadyResponse(_message.Message):
     __slots__ = ("ready",)