torchvision and opt_einsum not installed by requirements.txt

[jacksonthall22]

I get an error after running the installation instructions and trying the -h command:

python train.py spacetimeformer <any dataset> -h

Full tracebacks

Traceback (most recent call last):
  File "train.py", line 11, in <module>
    import spacetimeformer as stf
  File "c:\...\spacetimeformer\spacetimeformer\__init__.py", line 1, in <module>
    from . import data
  File "c:\...\spacetimeformer\spacetimeformer\data\__init__.py", line 7, in <module>
    from . import image_completion
  File "c:\...\spacetimeformer\spacetimeformer\data\image_completion\__init__.py", line 1, in <module>
    from .mnist import MNISTDset
  File "c:\...\spacetimeformer\spacetimeformer\data\image_completion\mnist.py", line 3, in <module>
    import torchvision
ModuleNotFoundError: No module named 'torchvision'

Then after pip install torchvision:

Traceback (most recent call last):
  File "train.py", line 11, in <module>
    import spacetimeformer as stf
  File "c:\...\spacetimeformer\spacetimeformer\__init__.py", line 14, in <module>
    from . import s4_model
  File "c:\...\spacetimeformer\spacetimeformer\s4_model\__init__.py", line 1, in <module>
    from .s4_model import S4_Forecaster
  File "c:\...\spacetimeformer\spacetimeformer\s4_model\s4_model.py", line 10, in <module>
    from . import s4_standalone
  File "c:\...\spacetimeformer\spacetimeformer\s4_model\s4_standalone.py", line 17, in <module>
    import opt_einsum as oe
ModuleNotFoundError: No module named 'opt_einsum'

Then after pip install opt_einsum the command runs as expected.