poly-lithic/pyproject.toml at main · ISISNeutronMuon/poly-lithic · 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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
[build-system]
requires = ["hatchling", "hatch-vcs", "wheel", "numpy<2.0"]
build-backend = "hatchling.build"

[tool.hatch.metadata]
allow-direct-references = true

[project]
name = "poly-lithic"
dynamic = ["version"]
description = "A package for deploying AI and other models and more using lume inspired interfaces"
authors = [
    {"name" = "Mateusz Leputa"},
    {"name" = "Kathryn Baker"},
    {"name" = "Mihnea Romanovschi"},
]
dependencies = [
    "pandas>=2.0.1",
    "numpy>=2.0.0",
    "mlflow==2.22.2",
    "boto3>=1.34.65",
    "psycopg2-binary",
    "k2eg>=0.3.0",
    "psutil>=6.0.0",
    "colorlog>=6.8.2",
    "p4p>=4.2.0",
    "pyepics>=3.5.7",
    "h5py>=3.12.1",
    "matplotlib>=3.10.1",
    "networkx>=3.4.2",
    "pydantic>=2.10.6",
    "pyyaml>=6.0.2",
    "requests>=2.32.3",
    "sympy>=1.13.1",
    "jinja2>=3.1.6",
    "yaspin>=3.2.0",
    "fastapi>=0.128.7",
    "uvicorn>=0.40.0",
    "httpx>=0.28.1"
]
[project.optional-dependencies]
dev = [
    "pytest",
    "pytest-asyncio",
    "pytest-xdist",
    "pytest-cov",
    "black",
    "flake8",
    "ruff",
    "isort",
    "pre-commit",
    "sphinx",
    "sphinx-rtd-theme",
    "twine",
    "wheel",
    "hatchling",
    "hatch-vcs",
    "torch>=2.6.0",
    "lume-model>=2.0.0",
]
torch = ["torch>=2.6.0", "lume-model>=2.0.0"]
tensorflow = ["keras>=3.1.1", "tensorflow>=2.14.0"]
tensorflow_cpu = ["keras>=3.1.1", "tensorflow-cpu>=2.14.0"]
lume = ["lume-model>=2.0.0"]
simframe-local = [
    # "simframe_services @ file:///home/gbm96348/nfs_home/simframe_services",
    # Please set the path to simframe_services via an environment variable or local editable install.
    # Example: "simframe_services @ file://${SIMFRAME_SERVICES_PATH}"
]


[tool.hatch.version]
source = "vcs" #test

[tool.hatch.build.hooks.vcs]
version-file = "poly_lithic/_version.py"

[tool.hatch.build.targets.sdist]
# exclude everying apart gthan the src directory
include =["poly_lithic"]

[tool.hatch.build.targets.wheel]
packages = ["poly_lithic"]

[tool.ruff.format]
preview = true
quote-style = "single"

[project.scripts]
poly_lithic = "poly_lithic.src.cli:cli"
pl = "poly_lithic.src.cli:cli"
model_manager = "poly_lithic.src.cli:cli" # legacy alias for poly_lithic


[tool.hatch.envs.default.env-vars]
PYTHONPATH = "poly_lithic/src"

[tool.hatch.envs.default.scripts]
test = "pytest"

[tool.hatch.version.raw-options]
local_scheme = "no-local-version"

[tool.hatch.scripts]
pre_install = "poly_lithic/pre_install.py"

[tool.hatch.envs.default.pipelines]
pre_install = ["pre_install"]

[[tool.hatch.envs.default.pipelines.install]]
hooks = ["pre_install"] # doesnt really work so will be moved in the future.

[tool.ruff.lint]
select = ["NPY201"]

# [tool.uv.sources]
# # NOTE: The following dependency uses a relative path to a sibling directory.
# # This will break in environments where '../pl-wrangler-simframe' does not exist.
# # Intended for local development only. For CI or production, install 'pl-wrangler-simframe' from PyPI or provide the directory.
# pl-wrangler-simframe = { path = "../pl-wrangler-simframe", editable = true }