[metadata]
name = torch_wrapper
version = 0.0.14
author = Repat
author_email = repat@repat.com
description = Wrapper for pytorch functions

[options]
package_dir = 
	= .
packages = find:
python_requires = >=3.8
install_requires = 
	numpy
	torch
	torchmetrics ~= 0.8
	aim ~= 3.10
	pytorch-lightning == 1.6.4

[options.extras_require]
dev = pytest

[options.packages.find]
include = torch_wrapper*

[egg_info]
tag_build = 
tag_date = 0