Fix tests

RasmussenLab · Nov 6, 2023 · 1af1ba5 · 1af1ba5
1 parent c460bdb
commit 1af1ba5
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/test/test_encode.py b/test/test_encode.py
@@ -157,10 +157,6 @@ def test_samples_too_small(self):
         t = self.tnfs.copy()
         l = self.lens.copy()
 
-        with self.assertWarns(UserWarning):
-            dl = vamb.encode.make_dataloader(r, t, l, batchsize=256)
-            vae.trainmodel(dl, batchsteps=None, nepochs=2)
-
     def test_loss_falls(self):
         vae = vamb.encode.VAE(self.rpkm.shape[1])
         rpkm_copy = self.rpkm.copy()