[metadata]
name = torch_wrapper
version = 0.0.26
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.9
	aim == 3.13.4
	pytorch-lightning == 1.7.7

[options.extras_require]
dev = pytest

[options.packages.find]
include = torch_wrapper*

[egg_info]
tag_build = 
tag_date = 0