dynomo import removed

dice-group · Nov 25, 2024 · a6e15b7 · a6e15b7
1 parent 4b1a876
commit a6e15b7
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/dicee/models/ensemble.py b/dicee/models/ensemble.py
@@ -1,11 +1,6 @@
 import torch
 import copy
 
-import torch._dynamo
-
-torch._dynamo.config.suppress_errors = True
-
-
 class EnsembleKGE:
     def __init__(self, seed_model):
         self.models = []