From 27d02aa6ee6b848b5ed5c5e74d3266ea4c0e432d Mon Sep 17 00:00:00 2001
From: dhritinaidu <dhritinaidu36@gmail.com>
Date: Fri, 12 Jul 2024 16:25:10 -0400
Subject: [PATCH] added pytest as a requirement

---
 requirements.txt  | 1 +
 src/test_local.py | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/requirements.txt b/requirements.txt
index af2b6e6..2cc6d29 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,5 +4,6 @@ numpy<2.0.0
 pylint
 pyinstaller
 google-api-python-client
+pytest
 torchvision
 torch==2.2.1
\ No newline at end of file
diff --git a/src/test_local.py b/src/test_local.py
index bdf7bb1..9187a1f 100644
--- a/src/test_local.py
+++ b/src/test_local.py
@@ -180,3 +180,5 @@ def test_infer_method(self):
 
 if __name__ == "__main__":
     unittest.main()
+
+