[build-system]
requires = ["setuptools"]
build-backend = "setuptools.build_meta"

[project]
name = "sarus_synthetic_data"
authors = [
    {name = "Sarus", email = "contact@sarus.tech"},
]
description = "package to train synthetic data generators"
readme = "README.md"
requires-python = ">=3.8"
license = {file = "PRIVATE"}
dynamic = ["version"]
classifiers = [
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.6",
    "Programming Language :: Python :: 3.7",
    "Programming Language :: Python :: 3.8"
]
dependencies = [
    "pandas~=1.4.0",
    "scipy>=1.5.0",
    "sarus-statistics>=4.0.0,<5.0.0",
    "pyarrow~=15.0",
    "jax==0.4.25",
    "jaxlib==0.4.25",
    "psutil",
    "tensorflow~=2.15.0",
    "flax~=0.8.1",
    "optax~=0.1.4",
    "orbax-checkpoint==0.4.3",
    "transformers~=4.43.0",
    "gcsfs",
    "pydantic~=2.7.0"
]

[tool.setuptools.packages.find]
exclude = ["tests"]
include = ["sarus_synthetic_data*"]

[project.optional-dependencies]
tests =[
	"pytest >= 6.2",
	"pytest-mock >= 3.6",
	"types-psutil"
]