[build-system]
requires = ["setuptools", "wheel", "protobuf==3.20.3", "mypy-protobuf==3.1.0"]
build-backend = "setuptools.build_meta"

[project]
name = "sarus_data_spec"
description = "A library to manage Sarus datasets"
readme = "README.md"
license = {text= "PRIVATE"}
requires-python = ">=3.8"
authors = [
    {name = "Sarus"}
]
classifiers =[
    "Programming Language :: Python :: 3.9",
]
dependencies = [
    "requests",
    'importlib-metadata; python_version<"3.10"',
    "protobuf == 3.20.3", # Due to https://github.com/awslabs/autogluon/issues/1725
	"numpy >= 1.20.0",
	"pyarrow ~= 15.0.0",
	"fsspec[http,gcs] >= 2021.0",# Generic FS support (e.g. in apache arrow)
	"pandas ~= 1.4.0",
	"typing_extensions >= 4.1.0",
]
dynamic = ["version"]

[tool.setuptools.packages.find]
exclude = ['tests','public','examples']
include = ['sarus_data_spec*']

[project.optional-dependencies]
tests =[
	"pytest >= 6.2",
	"pytest-mock >= 3.6",
	"pytest-cov >= 2.12",
	"pyspark",
    "psycopg2-binary",
	"docker",
	"types-cachetools",
    "types-requests",
    "types-setuptools",
    "types-python-dateutil",
    "mypy==1.9.0",
    "mypy-protobuf ==2.10.0",
    "sqlalchemy~=2.0",
    "pre-commit",
    "types-protobuf==3.18.4",
	"iso3166"
]

tensorflow = ["tensorflow >= 2.0; sys_platform != 'darwin' or (platform_machine != 'arm64')",
    "tensorflow-macos >=2.0; sys_platform == 'darwin' and (platform_machine == 'arm64')"]
onboarding = [
	"sarus-statistics>=4.0.1",
	"sarus-synthetic-data>=4.0.7",
	"clevercsv"]
external =["scikit-learn==1.2.2",
	"scipy >= 1.9.0",
	"shap==0.42.1",
	"imbalanced-learn",
	"scikit-optimize",
	"ydata-profiling<4.7.0", # this version is incompatible with visions.
	"visions",
	"plotly",
	"optbinning",
	"xgboost ~= 1.6.1"]
sql =[
	"pyqrlew>=0.9.26",
	]

bigquery = [
	"google-cloud",
	"google-cloud-dataproc",
	"google-cloud-bigquery",
	"google-cloud-bigquery-storage",
	"google-auth-stubs",
	"google-api-python-client-stubs",
	"sqlalchemy-bigquery"]

dpops =[
	"sarus-statistics>=4.0.0",
	"sarus-differential-privacy>=1.0.1",
]

llm = ["sarus-llm ~= 1.0.0"]