diff --git a/sci-ml/accelerate/accelerate-1.6.0-r2.ebuild b/sci-ml/accelerate/accelerate-1.6.0-r3.ebuild
similarity index 90%
rename from sci-ml/accelerate/accelerate-1.6.0-r2.ebuild
rename to sci-ml/accelerate/accelerate-1.6.0-r3.ebuild
index c1f01d3d43ec0..c96576336609f 100644
--- a/sci-ml/accelerate/accelerate-1.6.0-r2.ebuild
+++ b/sci-ml/accelerate/accelerate-1.6.0-r3.ebuild
@@ -29,9 +29,12 @@ RDEPEND="
 "
 DEPEND="${RDEPEND}"
 BDEPEND="test? (
-	dev-python/networkx
-	dev-python/parameterized
+	$(python_gen_cond_dep '
+		dev-python/networkx[${PYTHON_USEDEP}]
+		dev-python/parameterized[${PYTHON_USEDEP}]
+	')
 	sci-ml/caffe2[gloo]
+	sci-ml/torchdata[${PYTHON_SINGLE_USEDEP}]
 )"
 
 distutils_enable_tests pytest