diff --git a/dicee/models/ensemble.py b/dicee/models/ensemble.py
index e6c1e5de..83df1a3e 100644
--- 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 = []