From a8f8d55220380d1ece9d556d8463eabb68bf22c4 Mon Sep 17 00:00:00 2001 From: spencerwooo Date: Mon, 15 Apr 2024 17:04:47 +0800 Subject: [PATCH] Supplement PyPI fields --- README.md | 3 +++ pyproject.toml | 19 +++++++++++++++++++ src/torch_featurelayer/__init__.py | 2 +- 3 files changed, 23 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index cc739c9..99d46af 100644 --- a/README.md +++ b/README.md @@ -1,4 +1,7 @@ [![Ruff](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/charliermarsh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff) +[![image](https://img.shields.io/pypi/v/torch-featurelayer.svg)](https://pypi.python.org/pypi/torch-featurelayer) +[![image](https://img.shields.io/pypi/l/torch-featurelayer.svg)](https://pypi.python.org/pypi/torch-featurelayer) +[![image](https://img.shields.io/pypi/pyversions/torch-featurelayer.svg)](https://pypi.python.org/pypi/torch-featurelayer) [![lint](https://github.com/spencerwooo/torch-featurelayer/actions/workflows/ci.yml/badge.svg)](https://github.com/spencerwooo/torch-featurelayer/actions/workflows/ci.yml) # torch-featurelayer diff --git a/pyproject.toml b/pyproject.toml index dfbe872..1b2c741 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,11 +1,30 @@ [project] name = "torch-featurelayer" +description = "Useful utility functions and wrappers for hooking onto layers within PyTorch models for feature extraction." authors = [{ name = "spencerwooo", email = "spencer.woo@outlook.com" }] requires-python = ">=3.10,<3.13" readme = "README.md" +license = { file = "LICENSE" } +keywords = ["torch", "pytorch", "torchvision", "feature-extraction"] +classifiers = [ + "Development Status :: 5 - Production/Stable", + "Environment :: GPU :: NVIDIA CUDA", + "Intended Audience :: Developers", + "License :: OSI Approved :: MIT License", + "Operating System :: OS Independent", + "Programming Language :: Python :: 3.10", + "Programming Language :: Python :: 3.11", + "Programming Language :: Python :: 3.12", + "Programming Language :: Python :: 3 :: Only", + "Topic :: Software Development :: Libraries :: Python Modules", +] dependencies = ["torch>=2.0,<2.3"] dynamic = ["version"] +[project.urls] +Repository = "https://github.com/spencerwooo/torch-featurelayer" +Documentation = "https://github.com/spencerwooo/torch-featurelayer/blob/main/README.md" + [project.optional-dependencies] dev = ["mypy", "ruff"] test = ["torchvision>=0.15,<0.18"] diff --git a/src/torch_featurelayer/__init__.py b/src/torch_featurelayer/__init__.py index 468ee2e..885e918 100644 --- a/src/torch_featurelayer/__init__.py +++ b/src/torch_featurelayer/__init__.py @@ -2,5 +2,5 @@ from torch_featurelayer.feature_layers import FeatureLayers from torch_featurelayer.layer_candidates import get_layer_candidates -__version__ = '0.1.0' +__version__ = '0.1.1' __all__ = ['FeatureLayer', 'FeatureLayers', 'get_layer_candidates']