From 4fed3e01b3fa1fcf0dfe8313ad3f51cdb7d8c8e1 Mon Sep 17 00:00:00 2001
From: Daniel Smith <dgasmith@icloud.com>
Date: Fri, 6 Sep 2024 13:30:40 -0400
Subject: [PATCH] Fixes typo in

---
 opt_einsum/paths.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/opt_einsum/paths.py b/opt_einsum/paths.py
index f192664..5c38c45 100644
--- a/opt_einsum/paths.py
+++ b/opt_einsum/paths.py
@@ -500,9 +500,13 @@ def branch(
     size_dict: Dict[str, int],
     memory_limit: Optional[int] = None,
     nbranch: Optional[int] = None,
-    **optimizer_kwargs: Dict[str, Any],
+    cutoff_flops_factor: int = 4,
+    minimize: str = "flops",
+    cost_fn: str = "memory-removed",
 ) -> PathType:
-    optimizer = BranchBound(nbranch=nbranch, **optimizer_kwargs)  # type: ignore
+    optimizer = BranchBound(
+        nbranch=nbranch, cutoff_flops_factor=cutoff_flops_factor, minimize=minimize, cost_fn=cost_fn
+    )
     return optimizer(inputs, output, size_dict, memory_limit)