Merge pull request #16 from intsystems/NikitinaMaria-patch-1

Create requirements.txt

Author: NikitinaMaria

src/requirements.txt

@@ -0,0 +1,2 @@
+torch>=2.1.0+cu121
+torchvision>=0.16.0+cu121

src/setup.py

@@ -0,0 +1,34 @@
+import io
+import re
+from setuptools import setup, find_packages
+
+from irt import __version__
+
+def read(file_path):
+    with io.open(file_path, 'r', encoding='utf-8') as f:
+        return f.read()
+
+
+readme = read('README.rst')
+# вычищаем локальные версии из файла requirements (согласно PEP440)
+requirements = '\n'.join(
+    re.findall(r'^([^\s^+]+).*$',
+               read('requirements.txt'),
+               flags=re.MULTILINE))
+
+
+setup(
+    # metadata
+    name='irt',
+    version=__version__,
+    license='MIT',
+    author='Matvei Kreinin, Maria Nikitina, Petr Babkin, Iryna Zabarianska',
+    author_email="kreinin.mv@phystech.edu, nikitina.mariia@phystech.edu, babkin.pk@phystech.edu, akshiira@yandex.ru",
+    description='Implicit Reparametrization Trick',
+    long_description=readme,
+    url='https://github.com/intsystems/implicit-reparameterization-trick',
+
+    # options
+    packages=find_packages(),
+    install_requires=requirements,
+)