Skip to content

Commit

Permalink
add tbb to prevent numba crashes (#121)
Browse files Browse the repository at this point in the history
* add tbb to prevent numba crashes

* test whether `list[str]` is failing

* fix last lowercase type
  • Loading branch information
scottstanie authored Sep 6, 2023
1 parent 7a1f60d commit 8d46622
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
1 change: 1 addition & 0 deletions conda-env.yml
Original file line number Diff line number Diff line change
Expand Up @@ -20,4 +20,5 @@ dependencies:
- ruamel.yaml>=0.15
- scipy>=1.5 # "scipy 0.16+ is required for linear algebra", numba. 1.5 is the oldest version that supports Python 3.7
- shapely>=1.8
- tbb>=2021.6.0 # Module for numba threading which prevents `fork()` crashes
- threadpoolctl>=3.0
4 changes: 2 additions & 2 deletions src/dolphin/workflows/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -167,7 +167,7 @@ class UnwrapOptions(BaseModel, extra="forbid"):
)
_directory: Path = PrivateAttr(Path("unwrapped"))
unwrap_method: UnwrapMethod = UnwrapMethod.SNAPHU
tiles: list[int] = Field(
tiles: List[int] = Field(
[1, 1],
description=(
"Number of tiles to split the unwrapping into (for multi-scale unwrapping)."
Expand Down Expand Up @@ -210,7 +210,7 @@ class WorkerSettings(BaseModel, extra="forbid"):
" for wrapped-phase-estimation."
),
)
block_shape: tuple[int, int] = Field(
block_shape: Tuple[int, int] = Field(
(512, 512),
description="Size (rows, columns) of blocks of data to load at a time.",
)
Expand Down

0 comments on commit 8d46622

Please sign in to comment.