Merge branch 'MIC-DKFZ:main' into main

.gitlab-ci.yml

+default:
+  image: nvcr.io/nvidia/pytorch:22.01-py3
+
+stages:
+  - build
+
+build-job:
+  stage: build
+  script:
+    - echo "Check Versions"
+    - python --version
+    - python -c "import torch"
+    - pip --version
+    - pip uninstall torchtext -y
+    - echo "Install Package"
+    - pip install .
+    - pip install pytest
+    - echo "Batchgen Master"
+    - pip uninstall batchgenerators -y
+    - pip install git+https://github.com/MIC-DKFZ/batchgenerators.git
+    - echo "Run Tests"
+    - coverage run -m pytest -v
+    - coverage report
+    - coverage xml
+  artifacts:
+    reports:
+      cobertura: coverage.xml
+  coverage: '/TOTAL.*\s+(\d+\%)/'

setup.py

@@ -96,7 +96,7 @@ setup(
     name='nndet',
     version="v0.1",
     packages=find_packages(),
-    include_package_data=True,
+    # include_package_data=True,
     test_suite="unittest",
     long_description=readme,
     long_description_content_type='text/markdown',

tests/test_imports.py

+from nndet.utils.info import SuppressPrint
+
+
+def test_nevergrad_import():
+    import nevergrad as ng
+
+
+def test_batchgenerators_import():
+    import batchgenerators
+
+
+def test_pytorch_lightning_import():
+    import pytorch_lightning as pl
+
+
+def test_nnunet_import():
+    with SuppressPrint():
+        import nnunet.preprocessing.preprocessing as nn_preprocessing