scikit-llm

1
[build-system]
2
requires = ["setuptools>=61.0"]
3
build-backend = "setuptools.build_meta"
4

5
[project]
6
dependencies = [
7
  "scikit-learn>=1.1.0,<2.0.0",
8
  "pandas>=1.5.0,<3.0.0",
9
  "openai>=1.2.0,<2.0.0",
10
  "tqdm>=4.60.0,<5.0.0",
11
  "google-cloud-aiplatform[pipelines]>=1.27.0,<2.0.0"
12
]
13
name = "scikit-llm"
14
version = "1.0.0"
15
authors = [
16
  { name="Oleh Kostromin", email="kostromin97@gmail.com" },
17
  { name="Iryna Kondrashchenko", email="iryna230520@gmail.com" },
18
]
19
description = "Scikit-LLM: Seamlessly integrate powerful language models like ChatGPT into scikit-learn for enhanced text analysis tasks."
20
readme = "README.md"
21
license = {text = "MIT"}
22
requires-python = ">=3.9"
23
classifiers = [
24
    "Programming Language :: Python :: 3",
25
    "License :: OSI Approved :: MIT License",
26
    "Operating System :: OS Independent",
27
]
28

29
[project.optional-dependencies]
30
gpt4all = ["gpt4all>=2.0.0,<3.0.0"]
31
annoy = ["annoy>=1.17.2,<2.0.0"]
32

33
[tool.ruff]
34
select = [
35
    # pycodestyle
36
    "E",
37
    # pyflakes
38
    "F",
39
    # pydocstyle
40
    "D",
41
    # flake8-bandit
42
    "S",
43
    # pyupgrade
44
    "UP",
45
    # pep8-naming
46
    "N",
47
]
48
# Error E501 (Line too long) is ignored because of docstrings.
49
ignore = [
50
    "S101",
51
    "S301",
52
    "S311",
53
    "D100",
54
    "D200",
55
    "D203",
56
    "D205",
57
    "D401",
58
    "E501",
59
    "N803",
60
    "N806",
61
    "D104",
62
]
63
extend-exclude = ["tests/*.py", "setup.py"]
64
target-version = "py39"
65
force-exclude = true
66

67
[tool.ruff.per-file-ignores]
68
"__init__.py" = ["E402", "F401", "F403", "F811"]
69

70
[tool.ruff.pydocstyle]
71
convention = "numpy"
72

73
[tool.mypy]
74
ignore_missing_imports = true
75

76
[tool.black]
77
preview = true
78
target-version = ['py39', 'py310', 'py311']
79

80
[tool.isort]
81
profile = "black"
82
filter_files = true
83
known_first_party = ["skllm", "skllm.*"]
84
skip = ["__init__.py"]
85

86
[tool.docformatter]
87
close-quotes-on-newline = true # D209
88

89
[tool.pytest.ini_options]
90
pythonpath = [
91
  "."
92
]
93