Pipfile

[[source]]
name = "pypi"
url = "https://pypi.org/simple"
verify_ssl = true

[dev-packages]
black = ">=22.3.0"
flake8 = "*"
isort = "*"
nbqa = "*"
mypy = "*"
pre-commit = "*"
pytest = "*"
pytest-cov = "*"
pytest-mock = "*"
pytest-xdist = "*"
rich = "*"
watermark = "*"

[packages]
bambi = "*"
blackjax = "==0.9.6"
boto3 = "*"
botocore = "*"
catboost = "*"
causalml = "==0.12.3"
category-encoders = "*"
darts = "*"
dirty-cat = "*"
econml = "*"
feature-engine = "*"
graphviz = {version = "*", sys_platform = "== 'darwin'"}
hierarchicalforecast = "*"
imbalanced-learn = "*"
ipykernel = "*"
ipython = "*"
ipywidgets = "==7.7.2"
jax = "*"
jaxlib = "*"
jaxopt = "*"
lifelines = "*"
Lifetimes = "==0.11.3"
lightgbm = "*"
linearmodels = "*"
matplotlib = "*"
mlforecast = "*"
mlxtend = "*"
mistune = ">=2.0.3"
networkx = "*"
neuralforecast = "*"
nltk = "*"
numba = "*"
numpy = "*"
numpyro = "*"
openpyxl = "*"
optuna = "*"
orbit-ml = "*"
pandas = "*"
pandas-stubs = "*"
pystan = {git = "git+https://github.com/stan-dev/pystan2.git", ref = "v2.19.1.1"}
prophet = "==1.1.1"
pymc = "==4.2.2"
pytorch-lightning = "*"
seaborn = ">=0.12.0"
scikit-learn = "*"
scikit-survival = "*"
scikit-uplift = "*"
scipy = "*"
shap = "*"
skorch = "*"
sktime = "*"
spacy = "*"
spacy-transformers = "*"
statsmodels = "*"
tensorflow = "*"
tensorflow-probability = "*"
tfcausalimpact = "*"
transformers= "*"
torch = "*"
xgboost = "*"
yellowbrick= "*"
mkl = {version = "*", sys_platform = "== 'darwin'"}
intel-openmp = {version = "*", sys_platform = "== 'darwin'"}
tbb = {version = "*", sys_platform = "== 'darwin'"}

[requires]
python_version = "3.9"