fix: rounding to avoid !capacity contract error

graphprotocol · Oct 19, 2022 · b16af80 · b16af80
1 parent 672dc0b
commit b16af80
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/src/AllocationOpt.jl b/src/AllocationOpt.jl
@@ -161,7 +161,7 @@ function apply_preferences(
 end
 
 """
-    function optimize_indexer(indexer, repo, maximum_new_allocations, pinnedlist)
+    function optimize_indexer(indexer, repo, maximum_new_allocations,τ, filter_function, pinnedlist)
 
 # Arguments
 - `indexer::Indexer`: The indexer being optimised.
@@ -200,7 +200,7 @@ function optimize_indexer(
 
     # Filter results with deployment IPFS hashes
     suggested_allocations = Dict(
-        ipfshash(k) => v for (k, v) in zip(repo.subgraphs, ω) if v > 0.0
+        ipfshash(k) => round(v; digits=3) for (k, v) in zip(repo.subgraphs, ω) if v > 0.0
     )
     return suggested_allocations
 end

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -60,11 +60,11 @@ using GraphQLClient
             id, network_id, String[ipfshash], String[], String[], String[], gateway_url
         )
         # Check stake
-        @test isapprox(stake, optindexer.stake; atol=1e-6)
+        @test isapprox(stake, optindexer.stake; atol=1e-3)
         allocs = optimize_indexer(optindexer, repo, repo, 2, τ, filter_fn, String[])
         # Sum allocation amounts
         ω = sum(values(allocs))
-        @test isapprox(ω, stake; atol=1e-6)
+        @test isapprox(ω, stake; atol=1e-3)
         # Length of allocations is 1
         @test length(allocs) == 1
         # run optimize_indexer
@@ -80,7 +80,7 @@ using GraphQLClient
         allocs = optimize_indexer(optindexer, repo, repo, 2, τ, filter_fn, String[])
         # Sum allocation amounts
         ω = sum(values(allocs))
-        @test isapprox(ω, stake; atol=1e-6)
+        @test isapprox(ω, stake; atol=1e-3)
         # Length of allocations ≤ length of whitelist
         @test length(allocs) ≤ 2
 
@@ -96,7 +96,7 @@ using GraphQLClient
         @test pinned_allocs[another_ipfshash] == 0.1
         # Sum allocation amounts still satisfy stake constraint
         ω = sum(values(pinned_allocs))
-        @test isapprox(ω, stake; atol=1e-6)
+        @test isapprox(ω, stake; atol=1e-3)
 
         # run optimize_indexer with pinnedlist when pinned subgraph is optimal
         pinnedlist = String[ipfshash]
@@ -116,7 +116,7 @@ using GraphQLClient
         @test isapprox(pinned_allocs[ipfshash], allocs[ipfshash]; atol=1e-9)
         # Sum allocation amounts still satisfy stake constraint
         ω = sum(values(pinned_allocs))
-        @test isapprox(ω, stake; atol=1e-6)
+        @test isapprox(ω, stake; atol=1e-3)
     end
 
     @testset "read_filterlists" begin