X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pysvrt.git;a=blobdiff_plain;f=vignette_set.py;h=0ed3d39dcacbf5d5d52f97a74d10357a3c402bdb;hp=ea5215944bddbf7dcf0d5b50efcd20293f76da6a;hb=ea951479345890206211764657ce4d9556af9e76;hpb=c8bc2db12cddbf90b851798cd101632e7e9511ba

diff --git a/vignette_set.py b/vignette_set.py
index ea52159..0ed3d39 100755
--- a/vignette_set.py
+++ b/vignette_set.py
@@ -84,7 +84,7 @@ class CompressedVignetteSet:
             self.input_storages.append(svrt.compress(input.storage()))
 
         self.mean = acc / self.nb_batches
-        self.std = math.sqrt(acc_sq / self.nb_batches - self.mean * self.mean)
+        self.std = sqrt(acc_sq / self.nb_batches - self.mean * self.mean)
 
     def get_batch(self, b):
         input = torch.ByteTensor(svrt.uncompress(self.input_storages[b])).float()