test

app.py

@@ -74,6 +74,8 @@ from torch.utils.data import DataLoader
 import kornia
 # import warp as wp
 
+from pgnd.ffmpeg import make_video
+
 # print(torch.__version__)
 # print(torch.version.cuda)
 
@@ -469,6 +471,9 @@ class DynamicsVisualizer:
         self.config['far'] = far
     
     def init_model(self, batch_size, num_steps, num_particles, ckpt_path=None):
+        from pgnd.sim import Friction, CacheDiffSimWithFrictionBatch, StaticsBatch, CollidersBatch
+        from pgnd.material import PGNDModel
+
         self.cfg.sim.num_steps = num_steps
         cfg = self.cfg
 
@@ -908,12 +913,6 @@ class DynamicsVisualizer:
         # from torch.utils.data import DataLoader
         # import kornia
         import warp as wp
-        
-        from pgnd.sim import Friction, CacheDiffSimWithFrictionBatch, StaticsBatch, CollidersBatch
-        from pgnd.material import PGNDModel
-        from pgnd.utils import Logger, get_root, mkdir
-        from pgnd.ffmpeg import make_video
-        
 
         from diff_gaussian_rasterization import GaussianRasterizer
         from diff_gaussian_rasterization import GaussianRasterizationSettings as Camera