diff --git a/sci-ml/torchdata/Manifest b/sci-ml/torchdata/Manifest new file mode 100644 index 0000000000000..85b475dbee1a5 --- /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 0000000000000..4fd0b2861235d --- /dev/null +++ b/sci-ml/torchdata/metadata.xml @@ -0,0 +1,11 @@ + + + + + tupone@gentoo.org + Alfredo Tupone + + + pytorch/data + + 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 0000000000000..dfc4b19af95c3 --- /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 +}