diff --git a/qucumber/utils/training_statistics.py b/qucumber/utils/training_statistics.py
index d1a55135..6c5f2d73 100644
--- a/qucumber/utils/training_statistics.py
+++ b/qucumber/utils/training_statistics.py
@@ -14,7 +14,6 @@
 
 
 import torch
-from torch.nn import functional as F
 from torch.distributions.utils import probs_to_logits
 
 import qucumber.utils.cplx as cplx