setup.cfg

##########################
# Setup.py Configuration #
##########################
# Configuring setup()
[metadata]
name = pykeen
version = attr: src.pykeen.version.VERSION
description = A package for training and evaluating multimodal knowledge graph embeddings
long_description = file: README.md
long_description_content_type = text/markdown

# Links
url = https://github.com/pykeen/pykeen
download_url = https://github.com/pykeen/pykeen/releases
project_urls =
    Bug Tracker = https://github.com/pykeen/pykeen/issues
    Documentation = https://pykeen.readthedocs.io
    Homepage = https://pykeen.github.io

# Author information
author = "Mehdi Ali, Max Berrendorf, Charles Tapley Hoyt, Laurent Vermue"
author_email = pykeen2019@gmail.com
maintainer = "Mehdi Ali, Max Berrendorf, Charles Tapley Hoyt, Laurent Vermue"
maintainer_email = pykeen2019@gmail.com

# License information
license = MIT
license_file = LICENSE

# Search tags
classifiers =
    Development Status :: 4 - Beta
    Environment :: Console
    Intended Audience :: Science/Research
    License :: OSI Approved :: MIT License
    Operating System :: OS Independent
    Programming Language :: Python
    Programming Language :: Python :: 3.7
    Programming Language :: Python :: 3.8
    Programming Language :: Python :: 3.9
    Programming Language :: Python :: 3 :: Only
    Topic :: Scientific/Engineering :: Artificial Intelligence
    Topic :: Scientific/Engineering :: Chemistry
    Topic :: Scientific/Engineering :: Bio-Informatics
    Topic :: Scientific/Engineering :: Mathematics
    Topic :: Scientific/Engineering :: Information Analysis
keywords =
    Knowledge Graph Embeddings
    Machine Learning
    Data Mining
    Linked Data

[options]
install_requires =
    dataclasses-json
    numpy
    scipy>=1.5.0
    click
    click_default_group
    sklearn
    torch>=1.10; platform_system != "Windows"
    tqdm
    requests
    optuna>=2.0.0
    pandas>=1.0.0
    tabulate
    more_click
    more_itertools
    pystow>=0.1.5
    docdata
    class_resolver>=0.0.12
    pyyaml
    rexmex

zip_safe = false
include_package_data = True
python_requires = >=3.7

# Where is my code
packages = find:
package_dir =
    = src

[options.packages.find]
where = src

[options.extras_require]
templating =
    jinja2
plotting =
    matplotlib
    seaborn
mlflow =
    mlflow>=1.8.0
wandb =
    wandb
neptune =
    neptune-client
tensorboard =
    tensorboard
transformers =
    transformers
tests =
    unittest-templates>=0.0.5
    coverage
    pytest
docs =
    sphinx
    sphinx-rtd-theme
    sphinx-click
    sphinx-autodoc-typehints
    sphinx_automodapi
    texext

[options.entry_points]
console_scripts =
    pykeen = pykeen.cli:main

pykeen.datasets =
    hetionet         = pykeen.datasets.hetionet:Hetionet
    biokg            = pykeen.datasets.biokg:BioKG
    conceptnet       = pykeen.datasets.conceptnet:ConceptNet
    countries        = pykeen.datasets.countries:Countries
    drkg             = pykeen.datasets.drkg:DRKG
    kinships         = pykeen.datasets.kinships:Kinships
    nations          = pykeen.datasets.nations:Nations
    fb15k            = pykeen.datasets.freebase:FB15k
    fb15k237         = pykeen.datasets.freebase:FB15k237
    cn3l             = pykeen.datasets.wk3l:CN3l
    ckg              = pykeen.datasets.ckg:CKG
    cskg             = pykeen.datasets.cskg:CSKG
    codexsmall       = pykeen.datasets.codex:CoDExSmall
    codexmedium      = pykeen.datasets.codex:CoDExMedium
    codexlarge       = pykeen.datasets.codex:CoDExLarge
    dbpedia50        = pykeen.datasets.dbpedia:DBpedia50
    ogbbiokg         = pykeen.datasets.ogb:OGBBioKG
    ogbwikikg        = pykeen.datasets.ogb:OGBWikiKG
    openbiolink      = pykeen.datasets.openbiolink:OpenBioLink
    openbiolinklq    = pykeen.datasets.openbiolink:OpenBioLinkLQ
    umls             = pykeen.datasets.umls:UMLS
    wk3l15k          = pykeen.datasets.wk3l:WK3l15k
    wk3l120k         = pykeen.datasets.wk3l:WK3l120k
    wn18             = pykeen.datasets.wordnet:WN18
    wn18rr           = pykeen.datasets.wordnet:WN18RR
    yago310          = pykeen.datasets.yago:YAGO310
    db100k           = pykeen.datasets.db100k:DB100K
    wd50kt           = pykeen.datasets.wd50k:WD50KT
    wikidata5m       = pykeen.datasets.wikidata5m:Wikidata5M

######################
# Doc8 Configuration #
# (doc8.ini)         #
######################
[doc8]
max-line-length = 120
ignore =
    D000

##########################
# Coverage Configuration #
# (.coveragerc)          #
##########################
[coverage:run]
branch = True
source = pykeen
omit =
    tests/*
    docs/*
    src/pykeen/cli/*
    src/pykeen/__main__.py

[coverage:paths]
source =
    src/pykeen
    .tox/*/lib/python*/site-packages/pykeen

[coverage:report]
show_missing = True
exclude_lines =
    def __str__
    def __repr__

########################
# Pytest Configuration #
# (pytest.ini)         #
########################
[tool:pytest]
addopts = --strict-markers
markers =
    slow: marks tests as slow (deselect with '-m "not slow"')

##########################
# Darglint Configuration #
##########################
[darglint]
docstring_style = sphinx
strictness = short