diff --git a/minimol/model.py b/minimol/model.py
index 27b41ff..d1ee7eb 100644
--- a/minimol/model.py
+++ b/minimol/model.py
@@ -21,9 +21,6 @@
 
 from torch_geometric.data import Batch
 
-import random
-import numpy as np
-
 
 class Minimol: