Skip to content

Commit 580a5cf

Browse files
author
shenb888
committed
Describe what you changed
1 parent 506a24e commit 580a5cf

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

install.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -431,6 +431,17 @@ def install_pytorch_connectomics(
431431
print_success(f"Core packages installed: {', '.join(to_install)}")
432432
else:
433433
print_success("All core packages already installed")
434+
print_info("Ensuring numpy and h5py are installed from conda-forge (force reinstall)...")
435+
code, _, stderr = run_command(
436+
f"conda install -n {env_name} -c conda-forge numpy h5py -y --force-reinstall",
437+
check=False,
438+
)
439+
if code != 0:
440+
print_warning("conda reinstall of numpy/h5py failed; please verify the environment manually")
441+
if stderr.strip():
442+
print_warning(stderr.strip())
443+
else:
444+
print_success("numpy and h5py verified via conda-forge")
434445

435446
# Group 2: Optional scientific packages (nice to have, but slow to install)
436447
optional_packages = ["scipy", "scikit-learn", "scikit-image", "opencv"]

0 commit comments

Comments
 (0)