setup.cfg

[pycodestyle]
ignore=E402,W504
max-line-length=119

[pydocstyle]
; Use numpy style
convention=numpy

[build_sphinx]
version = 0.8.0b1

[metadata]
name = econml
version = 0.8.0b1
author = Microsoft Corporation
description = This package contains several methods for calculating Conditional Average Treatment Effects
long_description = file: README.md
long_description_content_type = text/markdown
license = MIT
keywords = treatment-effect
url = https://github.com/Microsoft/EconML
project_urls =
    Bug Tracker=https://github.com/Microsoft/EconML/Issues
    Source Code=https://github.com/Microsoft/EconML
    Documentation=https://econml.azurewebsites.net/
classifiers =
    Programming Language :: Python :: 3.5
    Programming Language :: Python :: 3.6
    Programming Language :: Python :: 3.7
    Programming Language :: Python :: 3.8
    License :: OSI Approved :: MIT License
    Operating System :: OS Independent

[options]
packages = find_namespace:
setup_requires =
    pytest-runner
    sphinx < 3.2
    sphinx_rtd_theme
install_requires =
    numpy
    scipy != 1.4.0
    scikit-learn > 0.21.0
    keras < 2.4
    sparse
    tensorflow > 1.10, < 2.3
    joblib >= 0.13.0
    numba != 0.42.1
    statsmodels >= 0.9
    graphviz
    matplotlib < 3.1; python_version <= '3.5'
    matplotlib; python_version > '3.5'
    llvmlite < 0.32; python_version <= '3.5'
    pandas < 1.1
test_suite = econml.tests
tests_require =
    pytest
    pytest-xdist
    pytest-cov
    jupyter
    nbconvert < 6
    traitlets < 5; python_version <= '3.5'
    seaborn
    lightgbm
    dowhy

[options.extras_require]
automl =
    azureml-sdk[explain,automl] == 1.0.83
    azure-cli
    
; TODO: exclude tests?
[options.packages.find]
include =
    econml
    econml.*

[options.package_data]
; include all CSV files as data
* = *.csv

[tool:pytest]
addopts = --junitxml=junit/test-results.xml -n auto --strict --cov-config=setup.cfg --cov=econml --cov-report=xml
markers =
    slow
    notebook
    automl

; coverage configuration
[coverage:run]
omit = econml/tests/*
branch = True
; need to explicitly add support for multiprocessing for OrthoForest
concurrency =
    thread
    multiprocessing