Merge pull request #5 from edong6768/changes/0.1.5

Changes/0.1.5

Author: dongyeoplee2

README.md

@@ -32,7 +32,7 @@ pip install git+https://github.com/edong6768/Malet.git
 - ml-collections 0.1.0
 - seaborn 0.11.2
 
-## Documentation
+## Documentation **(🚨 Plan to move to Sphinx-based Read the docs in near future)**
 
 ### Contents
 

pyproject.toml

@@ -33,6 +33,7 @@ dependencies = [
     'matplotlib==3.7.0',
     'ml-collections==0.1.0',
     'seaborn==0.11.2',
+    'rich'==13.6.0,
 ]
 
 [project.scripts]

requirements.txt

@@ -4,5 +4,4 @@ pandas==2.0.3
 matplotlib==3.7.0
 ml-collections==0.1.0
 seaborn==0.11.2
-rich==13.6.0
-tqdm
+rich==13.6.0

src/malet/experiment.py

@@ -10,7 +10,7 @@
 
 import pandas as pd
 import numpy as np
-from tqdm import tqdm
+from rich.progress import track
 
 from absl import logging
 from ml_collections.config_dict import ConfigDict
@@ -670,7 +670,7 @@ def resplit_logs(exp_folder_path: str, target_split: int=1, save_backup: bool=Tr
         # resplitting nth split
         cfgs_temp = copy.deepcopy(configs)
         cfgs_temp.filter_iter(lambda i, _: i%target_split==n)
-        for cfg in tqdm(cfgs_temp, desc=f'split: {n}/{target_split}'):
+        for cfg in track(cfgs_temp, description=f'split: {n}/{target_split}'):
           if cfg in base:
             metric_dict, info_dict = base.get_metric_and_info(cfg)
             lgs.add_result(cfg, metric_dict, **info_dict)