fix(test): lower down sample size for test

Novartis · Aug 12, 2024 · 4c86c1b · 4c86c1b
1 parent 7363753
commit 4c86c1b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scar/test/test_scar.py b/scar/test/test_scar.py
@@ -22,7 +22,7 @@ def test_scar(self):
             feature_type="sgRNAs",
         )
 
-        scarObj.train(epochs=40, batch_size=64)
+        scarObj.train(epochs=40, batch_size=32)
 
         scarObj.inference()
 
@@ -58,7 +58,7 @@ def test_scar_citeseq(self):
             feature_type="ADTs",
         )
 
-        citeseq_scar.train(epochs=200, batch_size=64, verbose=False)
+        citeseq_scar.train(epochs=200, batch_size=32, verbose=False)
         citeseq_scar.inference()
 
         dist = euclidean_distances(