fix: bump dependencies

kpn · Oct 9, 2024 · 908f142 · 908f142
1 parent 782e0cd
commit 908f142
Show file tree

Hide file tree

Showing 4 changed files with 679 additions and 644 deletions.
diff --git a/kstreams/prometheus/monitor.py b/kstreams/prometheus/monitor.py
@@ -132,7 +132,7 @@ def clean_stream_consumer_metrics(self, consumer: Consumer) -> None:
             partition = topic_partition.partition
 
             metrics_found = False
-            for sample in self.MET_LAG.collect()[0].samples:
+            for sample in list(self.MET_LAG.collect())[0].samples:
                 if {
                     "topic": topic,
                     "partition": str(partition),

diff --git a/kstreams/streams.py b/kstreams/streams.py
@@ -428,6 +428,13 @@ async def __anext__(self) -> ConsumerRecord:
         except errors.ConsumerStoppedError:
             raise StopAsyncIteration  # noqa: F821
 
+    async def __call__(self, *args, **kwargs) -> typing.Any:
+        """Make stream work as the wrapped func to keep behavior.
+
+        This allows for testing the stream as a normal function.
+        """
+        return self.func(*args, **kwargs)
+
 
 def stream(
     topics: typing.Union[typing.List[str], str],