From 784d45ab985974d78f69434628dd1e6e2c03991a Mon Sep 17 00:00:00 2001
From: MengW <1078137068@qq.com>
Date: Wed, 1 May 2024 22:54:38 +0800
Subject: [PATCH] perf: perfect code

---
 .github/workflows/unittest.yml | 28 ++++++++++++----------------
 CHANGELOG.md                   |  1 -
 2 files changed, 12 insertions(+), 17 deletions(-)

diff --git a/.github/workflows/unittest.yml b/.github/workflows/unittest.yml
index ee0c551..f48974e 100644
--- a/.github/workflows/unittest.yml
+++ b/.github/workflows/unittest.yml
@@ -18,22 +18,18 @@ jobs:
       - name: test_base
         run: |
           git clone https://github.com/NewT123-WM/tnlearn
-          cd tnlearn
-          pip install h5py~=3.10.0
-          pip install numpy~=1.26.2
-          pip install torch~=2.1.0
-          pip install sympy~=1.12
-          pip install setuptools~=68.0.0
-          pip install scikit-learn~=1.4.0
-          pip install scipy~=1.12.0
-          pip install joblib~=1.3.2
-          pip install requests~=2.31.0
-          pip install networkx~=3.2.1
-          pip install matplotlib~=3.8.3
-          pip install pandas~=2.2.0
-          pip install packaging~=23.2
-          pip install ipython~=8.18.1
-          pip install tqdm~=4.66.2
+          cd tnlearn 
+          pip install torch>=2.1.0
+          pip install numpy>=1.26.4
+          pip install scikit-learn>=1.4.1.post1
+          pip install pandas>=2.2.1
+          pip install sympy>=1.12
+          pip install tqdm>=4.66.2
+          pip install matplotlib>=3.8.3
+          pip install ipython>=8.18.1
+          pip install torchinfo>=1.8.0
+          pip install h5py>=3.10.0
+          pip install setuptools>=68.2.2
           python -m unittest
           
       
diff --git a/CHANGELOG.md b/CHANGELOG.md
index c2273ff..49d7a47 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,7 +1,6 @@
 ##  (2024-05-01)
 
 * perf: add examples ([fd06974](https://github.com/NewT123-WM/tnlearn/commit/fd06974))
-* perf: perf code comments ([9ed18a6](https://github.com/NewT123-WM/tnlearn/commit/9ed18a6))