From 5e01a89e2c41f1b63e4030bc1370d33e76b37182 Mon Sep 17 00:00:00 2001
From: sarthakpati <sarthak.pati@hotmail.com>
Date: Wed, 15 Jan 2025 15:26:38 -0500
Subject: [PATCH] add package data and data files to setup.py for requirements
 management

---
 setup.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/setup.py b/setup.py
index 8a67bcb2c5..1300af5908 100644
--- a/setup.py
+++ b/setup.py
@@ -72,8 +72,12 @@ def parse_args() -> argparse.Namespace:
         keywords=["pytorch", "torch", "training", "tools", "utilities"],
         python_requires=">=3.7",
         install_requires=read_requirements("requirements.txt"),
+        include_package_data=True,
         packages=find_packages(),
         package_data={"torchtnt": ["py.typed"]},
+        data_files=[
+            ('requirement_files',['requirements.txt', 'dev-requirements.txt']),
+        ],
         zip_safe=True,
         classifiers=[
             "Development Status :: 2 - Pre-Alpha",