fix extra parameter (#101) (#109)

* fix extra parameter in on_rebalance call (#101)

Author: bobh66

faust/__init__.py

@@ -23,7 +23,7 @@
 import typing
 from typing import Any, Mapping, NamedTuple, Optional, Sequence, Tuple
 
-__version__ = "0.5.1"
+__version__ = "0.5.2"
 __author__ = "Robinhood Markets, Inc."
 __contact__ = "[email protected], [email protected]"
 __homepage__ = "https://github.com/faust-streaming/faust"

faust/tables/base.py

@@ -568,9 +568,7 @@ async def on_rebalance(
         generation_id: int = 0,
     ) -> None:
         """Call when cluster is rebalancing."""
-        await self.data.on_rebalance(
-            self, assigned, revoked, newly_assigned, generation_id
-        )
+        await self.data.on_rebalance(assigned, revoked, newly_assigned, generation_id)
 
     async def on_recovery_completed(
         self, active_tps: Set[TP], standby_tps: Set[TP]

tests/unit/tables/test_base.py

@@ -559,7 +559,7 @@ async def test_on_rebalance(self, *, table):
         generation_id = 1
         await table.on_rebalance({TP1}, set(), set(), generation_id)
         table._data.on_rebalance.assert_called_once_with(
-            table, {TP1}, set(), set(), generation_id
+            {TP1}, set(), set(), generation_id
         )
 
     @pytest.mark.asyncio