From 6efe01d5fff9626829845ea994da1b42d36c7c76 Mon Sep 17 00:00:00 2001
From: thinhlpg <thinhlpg@gmail.com>
Date: Thu, 3 Apr 2025 16:51:18 +0700
Subject: [PATCH] chore: update Makefile and requirements for testing

- Added a 'test' target in Makefile to run unit tests using pytest.
- Included 'wandb' in requirements.txt for experiment tracking.
---
 Makefile         | 6 +++++-
 requirements.txt | 3 ++-
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/Makefile b/Makefile
index aba6059..2b8e2ce 100644
--- a/Makefile
+++ b/Makefile
@@ -1,10 +1,14 @@
-.PHONY: style quality install tensorboard clean fix update-worklog
+.PHONY: style quality install tensorboard clean fix update-worklog test
 
 # make sure to test the local checkout in scripts and not the pre-installed one
 export PYTHONPATH = src 
 
 check_dirs := . src notebooks scripts tests
 
+# Run tests
+test:
+	pytest -v tests/
+
 # Development dependencies
 install:
 	python -m venv venv && . venv/bin/activate && pip install --upgrade pip
diff --git a/requirements.txt b/requirements.txt
index 3888064..8c6581d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -15,4 +15,5 @@ python-dotenv
 loguru
 gradio
 tensorboard
-pytest
\ No newline at end of file
+pytest
+wandb
\ No newline at end of file