convert-labse-tf-pt/pyproject.toml at main · setu4993/convert-labse-tf-pt · 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
56
57
58
59
60
61
62
63
64
65
66
67
[tool.poetry]
name = "convert-labse-tf-pt"
version = "3.0.0"
description = "Convert LaBSE model from TensorFlow to PyTorch."
license = "MIT"
authors = ["Setu Shah <setu+labse@setu.me>"]
readme = "README.md"
homepage = "https://github.com/setu4993/convert-labse-tf-pt"
repository = "https://github.com/setu4993/convert-labse-tf-pt"
keywords = ["transformers", "bert", "labse", "pytorch", "tensorflow"]
packages = [{ include = "convert_labse_tf_pt", from = "src" }]
include = ["LICENSE"]

[tool.poetry.dependencies]
python = "^3.9,<3.11"
transformers = { extras = [
    "flax",
    "sentencepiece",
    "torch",
    "tf",
], version = "^4.34.1" }
tensorflow-hub = "^0.15.0"
ipywidgets = "^7.6.3"
loguru = "^0.6.0"
protobuf = "<5.0.0"
safetensors = "^0.3.1"

[tool.poetry.scripts]
convert_labse = "convert_labse_tf_pt.convert:convert_labse"
convert_lealla = "convert_labse_tf_pt.convert:convert_lealla"

[tool.poetry.group.test.dependencies]
pytest = "^7.2.0"
bert-for-tf2 = "^0.14.9"

[tool.poetry.group.dev.dependencies]
pre-commit = "^3.5.0"
ipython = "^8.7.0"

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

[tool.ruff]
line-length = 120
fix = true
unsafe-fixes = true
src = ["src"]

[tool.ruff.lint]
select = ["E4", "E7", "E9", "F", "I", "UP", "ASYNC"]
ignore = []

[tool.ruff.lint.isort]
known-first-party = ["galileo_core"]

[tool.ruff.lint.pydocstyle]
convention = "numpy"

[tool.docformatter]
style = "numpy"
recursive = true
wrap-summaries = 120
wrap-descriptions = 120

[tool.pytest.ini_options]
pythonpath = ["./src/"]