pyproject.toml

[build-system]
requires = ["poetry-core>=1.0.0", "poetry-dynamic-versioning"]
build-backend = "poetry_dynamic_versioning.backend"

[tool.poetry-dynamic-versioning]
enable = true
vcs = "git"
style = "semver"
pattern = "default-unprefixed"

[tool.poetry.dependencies]
python = "^3.8"

# Python lack of functionalities from future versions
typing-extensions = ">=3.10.0"

# Utility
packaging = "*"
click = ">=7.0"
setuptools = { version = "*", python = ">=3.12" }

# Networking
bravado = "^11.0.0"
oauthlib = ">=2.1.0"
PyJWT = "*"
requests = ">=2.20.0"
requests-oauthlib = ">=1.0.0"
websocket-client = ">=0.35.0, !=1.0.0"
urllib3 = "*"
swagger-spec-validator = ">=2.7.4"
protobuf = "^4.0.0"
neptune-api = "0.3.0"

# Built-in integrations
psutil = "*"
pandas = "*"

# Additional integrations
kedro-neptune = { version = "*", optional = true, python = ">=3.9,<3.12" }
neptune-detectron2 = { version = "*", optional = true}
neptune-fastai = { version = "*", optional = true }
neptune-lightgbm = { version = "*", optional = true }
pytorch-lightning = { version = "*", optional = true }
neptune-optuna = { version = "*", optional = true }
neptune-prophet = { version = "*", optional = true }
neptune-pytorch = { version = "*", optional = true }
neptune-sacred = { version = "*", optional = true }
neptune-sklearn = { version = "*", optional = true }
neptune-tensorflow-keras = { version = "*", optional = true }
neptune-tensorboard = { version = "*", optional = true }
neptune-xgboost = { version = "*", optional = true }
transformers = { version = "*", optional = true }
zenml = { version = "*", optional = true }
mosaicml = { version = "*", optional = true }
neptune-aws = { version = "*", optional = true }
neptune-mlflow = { version = "*", optional = true }
neptune-airflow = { version = "*", optional = true }

[tool.poetry.extras]
aws = ["neptune-aws"]
detectron2 = ["neptune-detectron2"]
fastai = ["neptune-fastai"]
kedro = ["kedro-neptune"]
lightgbm = ["neptune-lightgbm"]
optuna = ["neptune-optuna"]
prophet = ["neptune-prophet"]
pytorch = ["neptune-pytorch"]
pytorch-lightning = ["pytorch-lightning"]
sacred = ["neptune-sacred"]
sklearn = ["neptune-sklearn"]
tensorflow-keras = ["neptune-tensorflow-keras"]
tensorboard = ["neptune-tensorboard"]
transformers = ["transformers"]
xgboost = ["neptune-xgboost"]
zenml = ["zenml"]
mosaicml = ["mosaicml"]
mlflow = ["neptune-mlflow"]
airflow = ["neptune-airflow"]
experimental = ["neptune-experimental"]

[tool.poetry]
authors = ["neptune.ai <contact@neptune.ai>"]
description = "Neptune Client"
repository = "https://github.com/neptune-ai/neptune-client"
homepage = "https://neptune.ai/"
documentation = "https://docs.neptune.ai/"
include = ["CHANGELOG.md"]
license = "Apache License 2.0"
name = "neptune"
readme = "README.md"
version = "2.0.0"
classifiers = [
    "Development Status :: 5 - Production/Stable",
    "Environment :: Console",
    "Intended Audience :: Developers",
    "Intended Audience :: Science/Research",
    "License :: OSI Approved :: Apache Software License",
    "Natural Language :: English",
    "Operating System :: MacOS",
    "Operating System :: Microsoft :: Windows",
    "Operating System :: POSIX",
    "Operating System :: Unix",
    "Topic :: Software Development :: Libraries :: Python Modules",
    "Programming Language :: Python :: Implementation :: CPython",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
]
keywords = [
    "MLOps",
    "ML Experiment Tracking",
    "ML Model Registry",
    "ML Model Store",
    "ML Metadata Store",
]
packages = [
    { include = "neptune", from = "src" },
]

[tool.poetry.urls]
"Tracker" = "https://github.com/neptune-ai/neptune-client/issues"
"Documentation" = "https://docs.neptune.ai/"

[tool.poetry.scripts]
neptune = "neptune.cli.__main__:main"

[tool.black]
line-length = 120
target-version = ['py38', 'py39', 'py310', 'py311', 'py312']
include = '\.pyi?$,\_pb2\.py$'
exclude = '''
/(
    \.git
  | \.mypy_cache
  | \.tox
  | \.venv
  | build
  | dist
)/
'''

[tool.isort]
profile = "black"
line_length = 120
force_grid_wrap = 2

[tool.flake8]
max-line-length = 120
extend-ignore = "E203"

[tool.mypy]
files = 'src/neptune'
mypy_path = "stubs"
install_types = "True"
non_interactive = "True"
disallow_untyped_defs = "True"
disallow_any_unimported = "True"
no_implicit_optional = "True"
check_untyped_defs = "True"
warn_return_any = "True"
show_error_codes = "True"
warn_unused_ignores = "True"

[[tool.mypy.overrides]]
module = [
    "*.impl",
    "transformers.integrations",
    "composer.loggers",
    "pytorch_lightning.loggers",
]
ignore_missing_imports = "True"

[[tool.mypy.overrides]]
module = [
    "neptune.objects.abstract",
    "neptune.types.value_copy",
    "neptune.types.namespace",
    "neptune.types.series.series_value",
    "neptune.types.atoms.datetime",
    "neptune.types.atoms.boolean",
    "neptune.types.atoms.float",
    "neptune.types.atoms.integer",
    "neptune.types.atoms.string",
    "neptune.types.series.float_series",
    "neptune.types.series.string_series",
    "neptune.types.series.series",
    "neptune.types.sets.string_set",
    "neptune.attributes.atoms.boolean",
    "neptune.attributes.atoms.copiable_atom",
    "neptune.attributes.atoms.datetime",
    "neptune.attributes.atoms.float",
    "neptune.attributes.atoms.integer",
    "neptune.attributes.atoms.string",
    "neptune.attributes.attribute",
    "neptune.attributes.namespace",
    "neptune.attributes.series.fetchable_series",
    "neptune.attributes.series.float_series",
    "neptune.attributes.series.series",
    "neptune.attributes.series.string_series",
    "neptune.attributes.sets.string_set",
    "neptune.attributes.utils",
    "neptune.internal.exceptions",
    "neptune.internal.oauth",
    "neptune.internal.patches.bravado",
    "neptune.internal.utils.storage_utils",
    "neptune.internal.utils.utils",
    "neptune.internal.warnings",
    "neptune.internal.websockets.reconnecting_websocket",
    "neptune.internal.websockets.websocket_client_adapter",
    "neptune.exceptions",
    "neptune.handler",
    "neptune.integrations.python_logger",
    "neptune.integrations.utils",
    "neptune.internal.types.stringify_value",
    "neptune.internal.backends.api_model",
    "neptune.internal.backends.hosted_client",
    "neptune.internal.backends.hosted_neptune_backend",
    "neptune.internal.backends.neptune_backend",
    "neptune.internal.backends.neptune_backend_mock",
    "neptune.internal.backends.offline_neptune_backend",
    "neptune.internal.backends.operation_api_name_visitor",
    "neptune.internal.backends.operation_api_object_converter",
    "neptune.internal.backends.operations_preprocessor",
    "neptune.internal.backends.project_name_lookup",
    "neptune.internal.backends.swagger_client_wrapper",
    "neptune.internal.backends.utils",
    "neptune.internal.container_structure",
    "neptune.internal.credentials",
    "neptune.internal.id_formats",
    "neptune.internal.operation",
    "neptune.internal.streams.std_capture_background_job",
    "neptune.internal.streams.std_stream_capture_logger",
    "neptune.internal.daemon",
    "neptune.internal.utils.deprecation",
    "neptune.internal.utils.generic_attribute_mapper",
    "neptune.internal.utils.logger",
    "neptune.internal.utils.ping_background_job",
    "neptune.internal.utils.process_killer",
    "neptune.internal.utils.run_state",
    "neptune.internal.utils.uncaught_exception_handler",
    "neptune.internal.websockets.websocket_signals_background_job",
    "neptune.internal.websockets.websockets_factory",
    "neptune.objects.neptune_object",
    "neptune.objects.model",
    "neptune.objects.model_version",
    "neptune.objects.project",
    "neptune.objects.run",
    "neptune.objects.utils",
    "neptune.types.type_casting",
    "neptune.api.requests_utils",
    "neptune.internal.container_type",
    "neptune.internal.patches",
    "neptune.internal.utils",
]
ignore_errors = "True"