Pins MyPy

dgasmith · Sep 6, 2024 · 66ee53e · 66ee53e
1 parent 6996c6d
commit 66ee53e
Show file tree

Hide file tree

Showing 6 changed files with 8 additions and 7 deletions.
diff --git a/devtools/conda-envs/full-environment.yaml b/devtools/conda-envs/full-environment.yaml
@@ -16,7 +16,7 @@ dependencies:
 
     # Testing
   - codecov
-  - mypy
+  - mypy ==1.11*
   - pytest
   - pytest-cov
   - ruff ==0.5.*
diff --git a/devtools/conda-envs/min-deps-environment.yaml b/devtools/conda-envs/min-deps-environment.yaml
@@ -7,7 +7,7 @@ dependencies:
 
     # Testing
   - codecov
-  - mypy
+  - mypy ==1.11*
   - pytest
   - pytest-cov
-  - ruff ==0.5.*
+  - ruff ==0.6.*
diff --git a/devtools/conda-envs/min-ver-environment.yaml b/devtools/conda-envs/min-ver-environment.yaml
@@ -13,7 +13,7 @@ dependencies:
 
     # Testing
   - codecov
-  - mypy
+  - mypy ==1.11*
   - pytest
   - pytest-cov
   - ruff ==0.5.*
diff --git a/devtools/conda-envs/torch-only-environment.yaml b/devtools/conda-envs/torch-only-environment.yaml
@@ -11,7 +11,7 @@ dependencies:
 
     # Testing
   - codecov
-  - mypy
+  - mypy ==1.11*
   - pytest
   - pytest-cov
   - ruff ==0.5.*
diff --git a/docs/changelog.md b/docs/changelog.md
@@ -8,7 +8,7 @@ NumPy has been removed from `opt_einsum` as a dependancy allowing for more flexi
 **New Features**
 
  - [\#160](https://github.com/dgasmith/opt_einsum/pull/160) Migrates docs to MkDocs Material and GitHub pages hosting.
- - [\#161](https://github.com/dgasmith/opt_einsum/pull/161) Add type annotations to the code base.
+ - [\#161](https://github.com/dgasmith/opt_einsum/pull/161) Adds Python type annotations to the code base.
  - [\#204](https://github.com/dgasmith/opt_einsum/pull/204) Removes NumPy as a hard dependancy.
 
 **Enhancements**

diff --git a/opt_einsum/paths.py b/opt_einsum/paths.py
@@ -499,9 +499,10 @@ def branch(
     output: ArrayIndexType,
     size_dict: Dict[str, int],
     memory_limit: Optional[int] = None,
+    nbranch: Optional[int] = None,
     **optimizer_kwargs: Dict[str, Any],
 ) -> PathType:
-    optimizer = BranchBound(**optimizer_kwargs)  # type: ignore
+    optimizer = BranchBound(nbarnch=nbranch, **optimizer_kwargs)  # type: ignore
     return optimizer(inputs, output, size_dict, memory_limit)