X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pysvrt.git;a=blobdiff_plain;f=README.md;h=d0ec215a1767bd07bfc53b0e0758ce1891c39bd1;hp=b6e29c72db385f3c19cfba8bb04080313af98f6c;hb=77153ad6f6acb94a5132e9930722500cd93a6960;hpb=7966c9007cbcf9766b80781c927bf22fd20269af

diff --git a/README.md b/README.md
index b6e29c7..d0ec215 100644
--- a/README.md
+++ b/README.md
@@ -27,13 +27,13 @@ The svrtset.py implements the classes `VignetteSet` and
 `CompressedVignetteSet` with the following constructor
 
 ```
-def __init__(self, problem_number, nb_samples, batch_size, cuda = False, logger = None):
+__init__(problem_number, nb_samples, batch_size, cuda = False, logger = None)
 ```
 
 and the following method to return one batch
 
 ```
-def get_batch(self, b):
+(torch.FloatTensor, torch.LongTensor) get_batch(b)
 ```
 
 ## Low-level functions ##