From 42111a518ebb2258e073633c1e1caac25b51f546 Mon Sep 17 00:00:00 2001
From: NikitinaMaria <71255897+NikitinaMaria@users.noreply.github.com>
Date: Mon, 25 Nov 2024 22:08:14 +0300
Subject: [PATCH 1/2] Create requirements.txt

---
 src/requirements.txt | 2 ++
 1 file changed, 2 insertions(+)
 create mode 100644 src/requirements.txt

diff --git a/src/requirements.txt b/src/requirements.txt
new file mode 100644
index 0000000..7ea6faf
--- /dev/null
+++ b/src/requirements.txt
@@ -0,0 +1,2 @@
+torch>=2.1.0+cu121
+torchvision>=0.16.0+cu121

From 97ca22877712c63c1aa6eb81a4b96d5cacda1607 Mon Sep 17 00:00:00 2001
From: NikitinaMaria <71255897+NikitinaMaria@users.noreply.github.com>
Date: Mon, 25 Nov 2024 22:16:37 +0300
Subject: [PATCH 2/2] Create setup.py

---
 src/setup.py | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100644 src/setup.py

diff --git a/src/setup.py b/src/setup.py
new file mode 100644
index 0000000..502ab75
--- /dev/null
+++ b/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,
+)