RM device

jmschrei · Oct 31, 2024 · 63a644d · 63a644d
1 parent 6c1da9f
commit 63a644d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_seqlet.py b/tests/test_seqlet.py
@@ -159,7 +159,7 @@ def test_isotonic_thresholds_min_frac(X_contrib):
 
 
 def test_tfmodisco_seqlets(X_contrib):
-	seqlets = tfmodisco_seqlets(X_contrib, device='cpu')
+	seqlets = tfmodisco_seqlets(X_contrib)
 	pos_seqlets = seqlets[seqlets['attribution'] >= 0]
 	neg_seqlets = seqlets[seqlets['attribution'] < 0]