Remove FitVid requirement

s-tian · May 7, 2023 · f73e854 · f73e854
1 parent 3e51f04
commit f73e854
Show file tree

Hide file tree

Showing 3 changed files with 0 additions and 4 deletions.
diff --git a/environment.yml b/environment.yml
@@ -25,7 +25,6 @@ dependencies:
     - git+https://github.com/s-tian/iGibson.git@27216bf037436235814ec7a0843eb2b8a4e0376d
     - git+https://github.com/s-tian/robosuite.git@163e4d42866ecbb42c439fe0be4d5d52bb8320a1
     - git+https://github.com/s-tian/robomimic.git@1609a7ae4da4acc7aaab14ef731fdda82f5a7225
-    - git+ssh://git@github.com/s-tian/fitvid.git@9a8178056c022e5b2c8f22551ffab8510842db60
     - h5py==3.6.0
     - opencv-python==4.5.5.62
     - hydra-core==1.1.2

diff --git a/vp2/models/torch_fitvid_interface.py b/vp2/models/torch_fitvid_interface.py
@@ -4,7 +4,6 @@
 
 import pprint
 import torch
-import numpy as np
 from fitvid.model.fitvid import FitVid
 from vp2.mpc.utils import dict_to_numpy, slice_dict, cat_dicts
 from vp2.models.model import VideoPredictionModel

diff --git a/vp2/mpc/lbfgs.py b/vp2/mpc/lbfgs.py
@@ -1,9 +1,7 @@
 import numpy as np
 import torch
 import time
-from hydra.utils import instantiate
 
-from fitvid.utils.depth_utils import depth_to_rgb_im
 from vp2.mpc.optimizer import Optimizer
 from vp2.mpc.utils import ObservationList, write_moviepy_gif, dict_to_numpy