diff --git a/data analysis/analysis/analysis.py b/data analysis/analysis/analysis.py
index ce482fa9..9f33ac43 100644
--- a/data analysis/analysis/analysis.py	
+++ b/data analysis/analysis/analysis.py	
@@ -154,6 +154,7 @@ import numpy as np
 import math
 from sklearn import metrics
 from sklearn import preprocessing
+import torch
 
 class error(ValueError):
     pass