zeroshot-python/pyproject.toml at main · moonshinelabs-ai/zeroshot-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
[tool.poetry]
name = "zeroshot"
version = "0.1.11"
description = "Image classifier with zero-shot learning."
license = "MIT"

authors = [
    "Nate Harada <hello@usezeroshot.com>"
]
repository = "https://github.com/moonshinelabs/zeroshot-python"
homepage = "https://github.com/moonshinelabs/zeroshot-python"

readme = ["README.md"]
keywords = ["zeroshot", "classifier", "cv"]

# Setup dependencies.
[tool.poetry.dependencies]
python = ">=3.10,<3.13"
numpy = "^1.26.0"
pillow = "^10.0.1"
types-pillow = "^10.0.0.3"
mypy = "^1.6.0"
onnxruntime = "^1.16.1"
requests = "^2.31.0"
torch = {version = "^2.1.0", optional = true, source = "pytorch"}
xformers = {version = "^0.0.22.post7", optional = true, source = "pytorch"}

[[tool.poetry.source]]
name = "pytorch"
url = "https://download.pytorch.org/whl/cu121"
priority = "explicit"

[tool.poetry.extras]
torch = ["torch", "xformers"]

[tool.poetry.group.docs]
optional = true
[tool.poetry.group.docs.dependencies]
Sphinx = "^5.1.1"

[tool.poetry.group.dev.dependencies]
pytest = "^7.4.2"
parameterized = "^0.9.0"

# MyPy is missing stubs, which is annoying
scikit-learn = "^1.3.2"
requests-mock = "^1.11.0"
[[tool.mypy.overrides]]
module = [
    "parameterized",
    "pillow_heif",
    "requests",
    "onnxruntime"
]
ignore_missing_imports = true