The training command in Training Scripts doesn't work

[LingyuLi-math]

When I try to use the training command of '# d16, 256x256', it happens:

Traceback (most recent call last):
  File "/ssd/users/lingyu/Python/STFoundation/VAR/train.py", line 13, in <module>
    from utils import arg_util, misc
  File "/ssd/users/lingyu/Python/STFoundation/VAR/utils/arg_util.py", line 24, in <module>
    raise e
  File "/ssd/users/lingyu/Python/STFoundation/VAR/utils/arg_util.py", line 19, in <module>
    from tap import Tap
ModuleNotFoundError: No module named 'tap'

But I am sure that I have installed the package,

(py39) lingyu@woody:~/ssd/Python/STFoundation/VAR$ python
Python 3.9.18 (main, Sep 11 2023, 13:41:44) 
[GCC 11.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> from tap import Tap
>>> 

I found you mentioned typed-argument-parser, I also have tried:

try:
    from tap import Tap
except ImportError as e:
    print(f'`>>>>>>>> from tap import Tap` failed, please run:      pip3 install typed-argument-parser     <<<<<<<<', file=sys.stderr, flush=True)
    print(f'`>>>>>>>> from tap import Tap` failed, please run:      pip3 install typed-argument-parser     <<<<<<<<', file=sys.stderr, flush=True)
    time.sleep(5)
    raise e 

But it still cannot be run.