Merge pull request #589 from golemfactory/allow-total-amount-allocation

Allow total amount allocation
golemfactory · Aug 5, 2022 · 51e5b6f · 51e5b6f
2 parents 088d2db + 2c38b93
commit 51e5b6f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/goth/runner/probe/mixin.py b/goth/runner/probe/mixin.py
@@ -271,13 +271,13 @@ async def pay_invoices(
 
     @step()
     async def create_allocation(
-        self: ProbeProtocol, timeout: Optional[datetime] = None
+        self: ProbeProtocol, timeout: Optional[datetime] = None, total_amount=0
     ) -> Allocation:
         """Call create_allocation on the market api."""
 
         allocation = Allocation(
             allocation_id="",
-            total_amount=0,
+            total_amount=total_amount,
             spent_amount=0,
             remaining_amount=0,
             make_deposit=True,