test

app.py

@@ -68,10 +68,10 @@ zero.startup()
 # importlib.invalidate_caches()
 # torch = importlib.import_module("torch")
 
-# import torch
-# import torch.nn as nn
-# from torch.utils.data import DataLoader
-# import kornia
+import torch
+import torch.nn as nn
+from torch.utils.data import DataLoader
+import kornia
 # import warp as wp
 
 # print(torch.__version__)
@@ -903,17 +903,16 @@ class DynamicsVisualizer:
 
     @spaces.GPU
     def reset(self):
-        import torch
-        import torch.nn as nn
-        from torch.utils.data import DataLoader
-        import kornia
+        # import torch
+        # import torch.nn as nn
+        # 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