[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.2.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.types.type_casting",

"neptune.api.requests_utils",

"neptune.internal.container_type",

"neptune.internal.patches",

"neptune.internal.utils",

]

ignore_errors = "True"