pyproject.toml

[tool.poetry]
name = "keras-fsl"
version = "0.2.0"
description = "Model builders for some State-of-the-Art architectures in few-shot and contrastive learning"
authors = ["Clément Walter <clement0walter@gmail.com>"]

[tool.poetry.dependencies]
python = "^3.6.5"
tensorflow = "2.2.3"
tqdm = "^4.42.0"
tensorflow-probability = "^0.9.0"
protobuf = "~=3.19.0"

[tool.poetry.dev-dependencies]
pandas = "^1.0.1"
imgaug = "^0.4.0"
numpy = "^1.18.1"
pyyaml = "^5.3"
black = "^19.10b0"
pytest = "^5.4.1"
flake8 = "^3.7.9"
pytest-cov = "^2.8.1"
tensorflow-addons = "^0.9.1"
tensorflow-datasets = "^3.1.0"
seaborn = "^0.10.1"
gpumonitor = "^0.1.2"
jupyter = "^1.0.0"

[tool.black]
line-length = 127
target-version = ['py36', 'py37', 'py38']
include = '\.pyi?$'
exclude = '''
/(
    \.eggs
  | \.git
  | \venv
)/
'''

[build-system]
requires = ["poetry>=0.12"]
build-backend = "poetry.masonry.api"