[gentoo-commits] repo/gentoo:master commit in: sci-ml/torchdata/

[gentoo-commits] repo/gentoo:master commit in: sci-ml/torchdata/

[Alfredo Tupone]

commit:     64dc528a921700d875b6609378c28279d2b73ae2
Author:     Alfredo Tupone <tupone <AT> gentoo <DOT> org>
AuthorDate: Mon May 12 18:56:43 2025 +0000
Commit:     Alfredo Tupone <tupone <AT> gentoo <DOT> org>
CommitDate: Mon May 12 18:57:44 2025 +0000
URL:        https://gitweb.gentoo.org/repo/gentoo.git/commit/?id=64dc528a

sci-ml/torchdata: new package, add 0.11.0

Signed-off-by: Alfredo Tupone <tupone <AT> gentoo.org>

 sci-ml/torchdata/Manifest                |  1 +
 sci-ml/torchdata/metadata.xml            | 11 +++++++
 sci-ml/torchdata/torchdata-0.11.0.ebuild | 53 ++++++++++++++++++++++++++++++++
 3 files changed, 65 insertions(+)

diff --git a/sci-ml/torchdata/Manifest b/sci-ml/torchdata/Manifest
new file mode 100644
index 000000000000..85b475dbee1a
--- /dev/null
+++ b/sci-ml/torchdata/Manifest
@@ -0,0 +1 @@
+DIST torchdata-0.11.0.gh.tar.gz 1014016 BLAKE2B 8aa03cf6c928d5aaf87b31142b523e28b6c1486da10df72b96520ab8fd943ce71657e2d043f1a84aa6a4b550f349bbc4395b93f3a238f4e6833ec6b90cfa5337 SHA512 9a52fd1e16fa1b3e1fb50dd92ff322d4f664a3533c962fcbf7bbe9d7a4c51d0a858dc76b92633a1f75d138b263a8cca4b37db348f7c05991e73a9c702889a9de

diff --git a/sci-ml/torchdata/metadata.xml b/sci-ml/torchdata/metadata.xml
new file mode 100644
index 000000000000..4fd0b2861235
--- /dev/null
+++ b/sci-ml/torchdata/metadata.xml
@@ -0,0 +1,11 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE pkgmetadata SYSTEM "https://www.gentoo.org/dtd/metadata.dtd">
+<pkgmetadata>
+	<maintainer type="person">
+		<email>tupone@gentoo.org</email>
+		<name>Alfredo Tupone</name>
+	</maintainer>
+	<upstream>
+		<remote-id type="github">pytorch/data</remote-id>
+	</upstream>
+</pkgmetadata>

diff --git a/sci-ml/torchdata/torchdata-0.11.0.ebuild b/sci-ml/torchdata/torchdata-0.11.0.ebuild
new file mode 100644
index 000000000000..dfc4b19af95c
--- /dev/null
+++ b/sci-ml/torchdata/torchdata-0.11.0.ebuild
@@ -0,0 +1,53 @@
+# Copyright 2025 Gentoo Authors
+# Distributed under the terms of the GNU General Public License v2
+
+EAPI=8
+PYTHON_COMPAT=( python3_{11..13} )
+DISTUTILS_USE_PEP517=setuptools
+DISTUTILS_SINGLE_IMPL=1
+inherit distutils-r1
+
+DESCRIPTION="A repo for data loading and utilities on PyTorch domain libraries."
+HOMEPAGE="https://github.com/pytorch/data"
+SRC_URI="https://github.com/pytorch/data/archive/refs/tags/v${PV}.tar.gz
+	-> ${P}.gh.tar.gz"
+S="${WORKDIR}"/data-${PV}
+
+LICENSE="BSD"
+SLOT="0"
+KEYWORDS="~amd64"
+
+DEPEND="
+	$(python_gen_cond_dep '
+		dev-python/requests[${PYTHON_USEDEP}]
+		dev-python/urllib3[${PYTHON_USEDEP}]
+	')
+	sci-ml/pytorch[${PYTHON_SINGLE_USEDEP}]
+"
+RDEPEND="${DEPEND}"
+BDEPEND="
+	test? (
+		$(python_gen_cond_dep '
+			dev-python/expecttest[${PYTHON_USEDEP}]
+		')
+	)
+"
+
+distutils_enable_tests pytest
+
+src_test() {
+	local EPYTEST_DESELECT=(
+		test/stateful_dataloader/test_state_dict.py::TestMultiEpochSDL_shard0::test_multi_epoch_sdl_2
+		test/stateful_dataloader/test_state_dict.py::TestMultiEpochSDL_shard0::test_multi_epoch_sdl_3
+		test/stateful_dataloader/test_state_dict.py::TestEndOfEpochBehavior_shard0::test_end_of_epoch_behavior_2
+		test/stateful_dataloader/test_state_dict.py::TestEndOfEpochBehavior_shard0::test_end_of_epoch_behavior_3
+		test/stateful_dataloader/test_state_dict.py::TestMultiEpochState_shard0::test_pw
+		test/stateful_dataloader/test_state_dict.py::TestSingleIterCalled_shard0::test_mp
+		test/stateful_dataloader/test_state_dict.py::TestSingleIterCalled_shard0::test_mp_iter
+		test/stateful_dataloader/test_state_dict.py::TestStateInitializationDataset::test_mp
+		test/stateful_dataloader/test_state_dict.py::TestOutOfOrderWithCheckpointing::test_out_of_order_index_ds
+		test/stateful_dataloader/test_state_dict.py::TestOutOfOrderWithCheckpointing::test_out_of_order_iterable_ds_no_completed_workers
+		test/stateful_dataloader/test_state_dict.py::TestOutOfOrderWithCheckpointing::test_out_of_order_iterable_ds_one_completed_worker
+	)
+	distutils-r1_src_test
+}