environment.yml

# environment definition for development
# NOTE: versions are mostly unpinned, because this facilitates conda dependency resolution. Also,
# sensAI should be compatible to all (newer) versions of the dependencies. If it isn't, we need to specify 
# an upper version bound in setup.py (where it matters the most) as well as here.
# Compatibility with old (pinned) versions and the latest versions is tested in the GitHub CI build.
name: sensai
channels:
  - pytorch
  - defaults
  - conda-forge
dependencies:
  # basics
  - python=3.10
  - pip
  # optional sensai conda dependencies for development
  # NOTE: catboost is disabled, because it causes conflicts
  #- catboost
  - pytorch
  - pyqt
  - psutil
  - xgboost
  - lightgbm
  - utm=0.7.0
  - geopandas
  - sklearn-quantile
  # for notebook support
  - jupyter
  # for tests
  - pytest
  # for docs build
  - sphinx=5.0.2
  - sphinx_rtd_theme>=0.5.1
  - sphinx-toolbox=3.7.0
  - jupyter-book=0.15.1
  - nbsphinx
  # for release process
  - bump2version
  - pip:
      # non-optional sensai dependencies
      - -r requirements-relaxed.txt
      # optional sensai pip dependencies
      # clearml is only available through pip
      - clearml
      - pytorch-lightning
      - jsonargparse
      - docstring_parser
      # required for locally running the tox build (which will work on Linux only at this time)
      - tox
      - virtualenv