Skip to content

Commit

Permalink
Supplement PyPI fields
Browse files Browse the repository at this point in the history
  • Loading branch information
spencerwooo committed Apr 15, 2024
1 parent 7563e03 commit a8f8d55
Show file tree
Hide file tree
Showing 3 changed files with 23 additions and 1 deletion.
3 changes: 3 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
@@ -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
Expand Down
19 changes: 19 additions & 0 deletions pyproject.toml
Original file line number Diff line number Diff line change
@@ -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 = "[email protected]" }]
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"]
Expand Down
2 changes: 1 addition & 1 deletion src/torch_featurelayer/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -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']

0 comments on commit a8f8d55

Please sign in to comment.