| | |
| | |
| | |
| | |
| |
|
| | from pathlib import Path |
| | import re |
| | from typing import List, Tuple |
| |
|
| | from setuptools import setup, find_packages |
| |
|
| |
|
| | NAME = "dinov2" |
| | DESCRIPTION = "PyTorch code and models for the DINOv2 self-supervised learning method." |
| |
|
| | URL = "https://github.com/facebookresearch/dinov2" |
| | AUTHOR = "FAIR" |
| | REQUIRES_PYTHON = ">=3.9.0" |
| | HERE = Path(__file__).parent |
| |
|
| |
|
| | try: |
| | with open(HERE / "README.md", encoding="utf-8") as f: |
| | long_description = "\n" + f.read() |
| | except FileNotFoundError: |
| | long_description = DESCRIPTION |
| |
|
| |
|
| | def get_requirements(path: str = HERE / "requirements.txt") -> Tuple[List[str], List[str]]: |
| | requirements = [] |
| | extra_indices = [] |
| | with open(path) as f: |
| | for line in f.readlines(): |
| | line = line.rstrip("\r\n") |
| | if line.startswith("--extra-index-url "): |
| | extra_indices.append(line[18:]) |
| | continue |
| | requirements.append(line) |
| | return requirements, extra_indices |
| |
|
| |
|
| | def get_package_version() -> str: |
| | with open(HERE / "dinov2/__init__.py") as f: |
| | result = re.search(r"^__version__ = ['\"]([^'\"]*)['\"]", f.read(), re.M) |
| | if result: |
| | return result.group(1) |
| | raise RuntimeError("Can't get package version") |
| |
|
| |
|
| | requirements, extra_indices = get_requirements() |
| | version = get_package_version() |
| | dev_requirements, _ = get_requirements(HERE / "requirements-dev.txt") |
| | extras_requirements, _ = get_requirements(HERE / "requirements-extras.txt") |
| |
|
| |
|
| | setup( |
| | name=NAME, |
| | version=version, |
| | description=DESCRIPTION, |
| | long_description=long_description, |
| | long_description_content_type="text/markdown", |
| | author=AUTHOR, |
| | python_requires=REQUIRES_PYTHON, |
| | url=URL, |
| | packages=find_packages(), |
| | package_data={ |
| | "": ["*.yaml"], |
| | }, |
| | install_requires=requirements, |
| | extras_require={ |
| | "dev": dev_requirements, |
| | "extras": extras_requirements, |
| | }, |
| | dependency_links=extra_indices, |
| | install_package_data=True, |
| | license="Apache", |
| | license_files=("LICENSE",), |
| | classifiers=[ |
| | |
| | "Development Status :: 3 - Alpha", |
| | "Intended Audience :: Developers", |
| | "Intended Audience :: Science/Research", |
| | "License :: OSI Approved :: Apache Software License", |
| | "Programming Language :: Python :: 3.9", |
| | "Topic :: Scientific/Engineering :: Artificial Intelligence", |
| | "Topic :: Software Development :: Libraries :: Python Modules", |
| | ], |
| | ) |
| |
|
