pyproject.toml

[build-system]
requires = ["setuptools>=45", "setuptools_scm[toml]>=6.2"]
build-backend = "setuptools.build_meta"

[project]
name = "fd_shifts"
dynamic = ["version"]
requires-python = ">=3.11"
dependencies = [
  "albumentations>=1.0.3",
  "deepdiff",
  "faiss-cpu",
  "hydra-colorlog>=1.1.0",
  "hydra-core>=1.1.1",
  "hydra-zen",
  "imageio>=2.9.0",
  "ipython",
  "jsonargparse[signatures]>=4.29.0,<4.30.0",
  "loguru",
  "matplotlib>=3.3.4",
  "medmnist",
  "numpy>=1.22.2,<2.0.0",
  "ogb>=1.3.1",
  "omegaconf>=2.1.1",
  "opencv-python-headless",
  "pandarallel>=1.6.5",
  "pandas>=1.2.3",
  "Pillow>=9.5.0",
  "protobuf<=3.20.0",
  "pydantic>=1.10.0,<2.0.0",
  "lightning-bolts==0.7.0",
  "lightning==2.2.1",
  "rich>=10.7.0",
  "robustness @ https://github.com/MadryLab/robustness/archive/master.tar.gz",
  "scikit-image>=0.18.2",
  "scikit-learn>=0.24.2",
  "scipy>=1.6.1",
  "seaborn>=0.11.1",
  "shtab",
  "tensorboard>=2.4.1",
  "timm>=0.5.4",
  "toml>=0.10.2",
  "torch>=2.0.0",
  "torchmetrics>=0.2.0",
  "torchvision>=0.12.0",
  "tqdm>=4.62.0",
  "typing_extensions>=4.1.1",
  "wandb",
  "warmup_scheduler",
  "wilds>=1.1.0",
]

[project.optional-dependencies]
dev = [
  "pytest",
  "syrupy",
  "nbdime",
  "nbmake",
  "pylint",
  "black",
  "debugpy",
  "pre-commit",
]
docs = ["jupyterlab", "notebook", "ipykernel"]
launcher = ["parallel-ssh"]

[project.scripts]
fd-shifts = "fd_shifts.main:main"

[tool.setuptools_scm]
write_to = "fd_shifts/_version.py"

[tool.setuptools.packages]
find = {}

[tool.setuptools.package-data]
"*" = ["*.yaml"]

[tool.isort]
profile = "black"

[tool.black]
line-length = 88
exclude = '''

(
  /(
      \.git
    | \.venv
    | _build
  )/
)
'''

[tool.pyright]
exclude = [
    "experiments*",
    "analysis_outputs",
]
reportOptionalSubscript = "warning"
reportOptionalMemberAccess = "warning"
reportOptionalCall = "warning"
reportOptionalIterable = "warning"
reportOptionalContextManager = "warning"
reportOptionalOperand = "warning"
# useLibraryCodeForTypes = false

[tool.pytest.ini_options]
testpaths = [ "fd_shifts" ]
addopts = [
  "--import-mode=importlib",
]
xfail_strict = true
filterwarnings = [
  'ignore:fields may not start with an underscore:RuntimeWarning',
  # Ignore tensorboard deprecation warnings
  'ignore:distutils:DeprecationWarning',
  'ignore:Call to deprecated create function:DeprecationWarning',
  # Ignore timm deprecation warnings
  'ignore:.*is deprecated and will be removed in Pillow 10:DeprecationWarning',
  'ignore:torch.distributed._sharded_tensor will be deprecated, use torch.distributed._shard.sharded_tensor instead:DeprecationWarning',
  # Ignore lightning_bolts optional dependencies warnings
  'ignore:You want to use:UserWarning',
]
markers = [
    "baurc:selects all tests for this metric",
]

[tool.pylint]
extension-pkg-allow-list = [
  "numpy",
  "torch",
]
class-const-naming-style = "any"
load-plugins = ["pylint.extensions.docparams"]

# The type of string formatting that logging methods do. `old` means using %
# formatting, `new` is for `{}` formatting.
logging-format-style = "new"

# Logging modules to check that the string format arguments are in logging
# function parameter format.
logging-modules = ["logging", "loguru"]

disable = [
  "raw-checker-failed",
  "bad-inline-option",
  "locally-disabled",
  "file-ignored",
  "suppressed-message",
  "useless-suppression",
  "deprecated-pragma",
  "use-symbolic-message-instead",
  "missing-module-docstring",
]

max-args = 10
max-attributes = 20
ignore-paths = [
  "fd_shifts/models/networks/residual_flows/*",
  "fd_shifts/tests/*",
]
accept-no-param-doc = false

accept-no-raise-doc = false

accept-no-return-doc = false

accept-no-yields-doc = false

[tool.mypy]
# strict = true
plugins = [
  "pydantic.mypy",
]

show_error_codes = true
show_column_numbers = true
follow_imports = "silent"

exclude = [
  'fd_shifts/models/networks/residual_flows/.*',
  'fd_shifts/tests/.*',
]

files = [
  "fd_shifts",
]

python_version = "3.10"

warn_unused_configs = true
disallow_any_generics = true
disallow_subclassing_any = true
disallow_untyped_calls = true
disallow_untyped_defs = true
disallow_incomplete_defs = true
check_untyped_defs = true
disallow_untyped_decorators = false
no_implicit_optional = true
warn_redundant_casts = true
warn_unused_ignores = true
warn_return_any = true
no_implicit_reexport = true
strict_equality = true
strict_concatenate = true

[[tool.mypy.overrides]]
module = [
  "albumentations.*",
  "timm.*",
  "tensorflow.*",
  "tensorboard.*",
  "matplotlib.*",
  # "numpy.*",
  "multiprocessing.*",
  "torchvision.*",
  "torch.*",
  "scipy.*",
  "IPython.*",
  "skimage.*",
  # "mypy.*",
  "usort.*",
  "sklearn.*",
  "hydra_zen.*",
  "hydra_zen",
  "deepdiff.*",
  "cv2.*",
  "pl_bolts.*",
  "seaborn",
  "pssh.*",
  "wilds.*",
  "robustness.*",
]
ignore_missing_imports = true

[tool.pydantic-mypy]
init_forbid_extra = true
init_typed = true
warn_required_dynamic_aliases = true
warn_untyped_fields = true