From 628671a5bfe794f1f0f900975890155a4030c4a7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jofre=20Vall=C3=A8s?= <jofrevalles99@gmail.com>
Date: Fri, 3 Nov 2023 10:47:05 +0100
Subject: [PATCH] Fix KaHyPar test

---
 test/KaHyPar_test.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/test/KaHyPar_test.jl b/test/KaHyPar_test.jl
index 559e0ba..66ff887 100644
--- a/test/KaHyPar_test.jl
+++ b/test/KaHyPar_test.jl
@@ -41,10 +41,10 @@
             EinExpr([:a, :C, :d], Dict(:a => 3, :d => 6, :C => 4)),
         ]
 
-        path = einexpr(HyPar, EinExpr(Symbol[], tensors))
+        path = einexpr(HyPar(imbalance=0.45), EinExpr(Symbol[], tensors))
 
         @test path isa EinExpr
 
-        @test mapreduce(flops, +, Branches(path)) == 31653164
+        @test mapreduce(flops, +, Branches(path)) == 19099592
     end
 end
\ No newline at end of file