From 8f7ae775953781dfe8b71a407ee6da93111eace5 Mon Sep 17 00:00:00 2001
From: bxdd <bxddream@gmail.com>
Date: Sat, 22 Apr 2023 16:23:40 +0800
Subject: [PATCH] [MNT] Update setup

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

diff --git a/setup.py b/setup.py
index 106e033..9807b39 100644
--- a/setup.py
+++ b/setup.py
@@ -65,6 +65,7 @@ REQUIRED = [
     "lightgbm>=3.3.0",
     "psutil>=5.9.4",
     "torchvision>=0.15.1",
+    "tensorflow>=2.4.0",
 ]
 
 if get_platform() != MACOS: