Skip to content

Commit

Permalink
update meta.yaml
Browse files Browse the repository at this point in the history
  • Loading branch information
wolny committed Dec 7, 2023
1 parent 3e8f551 commit f59cc4e
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 7 deletions.
20 changes: 14 additions & 6 deletions conda-recipe/meta.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -7,21 +7,22 @@ source:

build:
noarch: python
script: {{ PYTHON }} -m pip install . -vv --no-deps --no-build-isolation --ignore-installed --no-cache-dir
number: 0
entry_points:
- predict3dunet = pytorch3dunet.predict:main
- train3dunet = pytorch3dunet.train:main

requirements:
build:
- python
- setuptools
- python >=3.9
- pip

run:
- python 3.9
- python >=3.9
- pytorch
- tensorboard
- tqdm
- setuptools
- h5py
- scipy
- scikit-image
Expand All @@ -35,9 +36,16 @@ test:
- pytorch3dunet.datasets
requires:
- pytest
run:
- py.test
source_files:
- tests
commands:
- pytest tests

about:
home: https://github.com/wolny/pytorch-3dunet
summary: 'Versatile U-Net implementation for volumetric semantic segmentation written in PyTorch'
description: |
PyTorch implementation of 3D U-Net and its variants (ResidualUNet3D, ResidualUNetSE3D).
The code allows for training the U-Net for both: semantic segmentation (binary and multi-class)
and regression problems (e.g. de-noising, learning deconvolutions).
license: MIT
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
author="Adrian Wolny, Lorenzo Cerrone",
url="https://github.com/wolny/pytorch-3dunet",
license="MIT",
python_requires='>=3.7',
python_requires='>=3.9',
entry_points={'console_scripts': [
'train3dunet=pytorch3dunet.train:main',
'predict3dunet=pytorch3dunet.predict:main']
Expand Down

0 comments on commit f59cc4e

Please sign in to comment.