torchwrench 0.7.0__tar.gz

torchwrench-0.7.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Labbeti
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

torchwrench-0.7.0/PKG-INFO

@@ -0,0 +1,306 @@
+Metadata-Version: 2.4
+Name: torchwrench
+Version: 0.7.0
+Summary: Collection of functions and modules to help development in PyTorch.
+Author-email: "Étienne Labbé (Labbeti)" <labbeti.pub@gmail.com>
+Maintainer-email: "Étienne Labbé (Labbeti)" <labbeti.pub@gmail.com>
+License: MIT License
+        
+        Copyright (c) 2025 Labbeti
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy
+        of this software and associated documentation files (the "Software"), to deal
+        in the Software without restriction, including without limitation the rights
+        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+        copies of the Software, and to permit persons to whom the Software is
+        furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all
+        copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+        SOFTWARE.
+        
+Project-URL: Homepage, https://pypi.org/project/torchwrench/
+Project-URL: Documentation, https://torchwrench.readthedocs.io/
+Project-URL: Repository, https://github.com/Labbeti/torchwrench.git
+Project-URL: Changelog, https://github.com/Labbeti/torchwrench/blob/main/CHANGELOG.md
+Project-URL: Tracker, https://github.com/Labbeti/torchwrench/issues
+Keywords: pytorch,deep-learning
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Scientific/Engineering
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: pythonwrench~=0.1.0
+Requires-Dist: torch>=1.10.0
+Requires-Dist: typing-extensions>=4.10.0
+Provides-Extra: dev
+Requires-Dist: coverage[toml]>=7.9.1; extra == "dev"
+Requires-Dist: flake8>=7.2.0; extra == "dev"
+Requires-Dist: ipython>=8.18.1; extra == "dev"
+Requires-Dist: pre-commit>=4.2.0; extra == "dev"
+Requires-Dist: pytest>=8.4.1; extra == "dev"
+Requires-Dist: ruff>=0.12.0; extra == "dev"
+Requires-Dist: setuptools>=80.9.0; extra == "dev"
+Requires-Dist: soundfile>=0.13.1; extra == "dev"
+Requires-Dist: sphinx>=7.4.7; extra == "dev"
+Requires-Dist: sphinx-press-theme>=0.8.0; extra == "dev"
+Requires-Dist: twine>=6.1.0; extra == "dev"
+Requires-Dist: typing-extensions>=4.10.0; extra == "dev"
+Requires-Dist: ipykernel>=6.29.5; extra == "dev"
+Requires-Dist: numpy!=2.0.0,!=2.0.1,!=2.0.2; extra == "dev"
+Provides-Extra: extras
+Requires-Dist: colorlog>=6.9.0; extra == "extras"
+Requires-Dist: h5py>=3.14.0; extra == "extras"
+Requires-Dist: numpy!=2.0.0,!=2.0.1,!=2.0.2; extra == "extras"
+Requires-Dist: omegaconf>=2.3.0; extra == "extras"
+Requires-Dist: pandas>=2.3.0; extra == "extras"
+Requires-Dist: pyyaml>=6.0.2; extra == "extras"
+Requires-Dist: safetensors>=0.5.3; extra == "extras"
+Requires-Dist: scipy>=1.13.1; extra == "extras"
+Requires-Dist: tensorboard>=2.19.0; extra == "extras"
+Requires-Dist: torchaudio>=0.10.0; extra == "extras"
+Requires-Dist: torchvision>=0.11.0; extra == "extras"
+Requires-Dist: tqdm>=4.67.1; extra == "extras"
+Dynamic: license-file
+
+# torchwrench
+
+<center>
+
+<a href="https://www.python.org/">
+    <img alt="Python" src="https://img.shields.io/badge/-Python 3.9+-blue?style=for-the-badge&logo=python&logoColor=white">
+</a>
+<a href="https://github.com/Labbeti/torchwrench/actions">
+    <img alt="Build" src="https://img.shields.io/github/actions/workflow/status/Labbeti/torchwrench/test.yaml?branch=main&style=for-the-badge&logo=github">
+</a>
+<a href='https://torchwrench.readthedocs.io/'>
+    <img src='https://readthedocs.org/projects/torchwrench/badge/?version=stable&style=for-the-badge' alt='Documentation Status' />
+</a>
+<a href="https://pytorch.org/get-started/locally/">
+    <img alt="PyTorch" src="https://img.shields.io/badge/-PyTorch 1.10+-ee4c2c?style=for-the-badge&logo=pytorch&logoColor=white">
+</a>
+
+Collection of functions and modules to help development in PyTorch.
+
+</center>
+
+
+## Installation
+
+With pip:
+```bash
+pip install torchwrench
+```
+
+With uv:
+```bash
+uv add torchwrench
+```
+
+The main requirement is **[PyTorch](https://pytorch.org/)**.
+
+To check if the package is installed and show the package version, you can use the following command in your terminal:
+```bash
+torchwrench-info
+```
+
+This library works on all Python versions **>=3.9**, all PyTorch versions **>= 1.10**, and on **Linux, Mac and Windows** systems.
+
+## Examples
+
+`torchwrench` functions and modules can be used like `torch` ones. The default acronym for `torchwrench` is `tw`.
+
+### Label conversions
+Supports **multiclass** labels conversions between probabilities, classes indices, classes names and onehot encoding.
+
+```python
+import torchwrench as tw
+
+probs = tw.as_tensor([[0.9, 0.1], [0.4, 0.6]])
+names = tw.probs_to_name(probs, idx_to_name={0: "Cat", 1: "Dog"})
+# ["Cat", "Dog"]
+```
+
+This package also supports **multilabel** labels conversions between probabilities, classes multi-indices, classes multi-names and multihot encoding.
+
+```python
+import torchwrench as tw
+
+multihot = tw.as_tensor([[1, 0, 0], [0, 1, 1], [0, 0, 0]])
+indices = tw.multihot_to_indices(multihot)
+# [[0], [1, 2], []]
+```
+
+Finally, this packages includes the **powerset multilabel** conversions :
+
+```python
+import torchwrench as tw
+
+multihot = tw.as_tensor([[1, 0, 0], [0, 1, 1], [0, 0, 0]])
+indices = tw.multilabel_to_powerset(multihot, num_classes=3, max_set_size=2)
+# tensor([[0, 1, 0, 0, 0, 0, 0],
+#         [0, 0, 0, 0, 0, 0, 1],
+#         [1, 0, 0, 0, 0, 0, 0]])
+```
+
+### Typing
+
+Typing with number of dimensions :
+
+```python
+import torchwrench as tw
+
+x1 = tw.as_tensor([1, 2])
+print(isinstance(x1, tw.Tensor2D))  # False
+x2 = tw.as_tensor([[1, 2], [3, 4]])
+print(isinstance(x2, tw.Tensor2D))  # True
+```
+
+Typing with tensor dtype :
+
+```python
+import torchwrench as tw
+
+x1 = tw.as_tensor([1, 2], dtype=tw.int)
+print(isinstance(x1, tw.SignedIntegerTensor))  # True
+
+x2 = tw.as_tensor([1, 2], dtype=tw.long)
+print(isinstance(x2, tw.SignedIntegerTensor1D))  # True
+
+x3 = tw.as_tensor([1, 2], dtype=tw.float)
+print(isinstance(x3, tw.SignedIntegerTensor))  # False
+```
+
+### Padding & cropping
+
+Pad a specific dimension :
+
+```python
+import torchwrench as tw
+
+x = tw.rand(10, 3, 1)
+padded = tw.pad_dim(x, target_length=5, dim=1, pad_value=-1)
+# x2 has shape (10, 5, 1), padded with -1
+```
+
+Pad nested list of tensors to a single one :
+
+```python
+import torchwrench as tw
+
+tensors = [tw.rand(10, 2), [tw.rand(3)] * 5, tw.rand(0, 5)]
+padded = tw.pad_and_stack_rec(tensors, pad_value=0)
+# padded has shape (3, 10, 5), padded with 0
+```
+
+Remove values at a specific dimension :
+
+```python
+import torchwrench as tw
+
+x = tw.rand(10, 5, 3)
+cropped = tw.crop_dim(x, dim=1, target_length=2)
+# cropped has shape (10, 2, 3)
+```
+
+### Masking
+
+```python
+import torchwrench as tw
+
+x = tw.as_tensor([3, 1, 2])
+mask = tw.lengths_to_non_pad_mask(x, max_len=4)
+# Each row i contains x[i] True values for non-padding mask
+# tensor([[True, True, True, False],
+#         [True, False, False, False],
+#         [True, True, False, False]])
+```
+
+```python
+import torchwrench as tw
+
+x = tw.as_tensor([1, 2, 3, 4])
+mask = tw.as_tensor([True, True, False, False])
+result = tw.masked_mean(x, mask)
+# result contains the mean of the values marked as True: 1.5
+```
+
+### Others tensors manipulations!
+
+```python
+import torchwrench as tw
+
+x = tw.as_tensor([1, 2, 3, 4])
+result = tw.insert_at_indices(x, indices=[0, 2], values=5)
+# result contains tensor with inserted values: tensor([5, 1, 2, 5, 3, 4])
+```
+
+```python
+import torchwrench as tw
+
+perm = tw.randperm(10)
+inv_perm = tw.get_inverse_perm(perm)
+
+x1 = tw.rand(10)
+x2 = x1[perm]
+x3 = x2[inv_perm]
+# inv_perm are indices that allow us to get x3 from x2, i.e. x1 == x3 here
+```
+
+### Extra: pre-compute datasets to HDF files
+
+Here is an example of pre-computing spectrograms of torchaudio `SPEECHCOMMANDS` dataset, using `pack_dataset` function:
+
+```python
+from torchaudio.datasets import SPEECHCOMMANDS
+from torchaudio.transforms import Spectrogram
+from torchwrench import nn
+from torchwrench.extras.hdf import pack_to_hdf
+
+speech_commands_root = "path/to/speech_commands"
+packed_root = "path/to/packed_dataset.hdf"
+
+dataset = SPEECHCOMMANDS(speech_commands_root, download=True, subset="validation")
+# dataset[0] is a tuple, contains waveform and other metadata
+
+class MyTransform(nn.Module):
+    def __init__(self) -> None:
+        super().__init__()
+        self.spectrogram_extractor = Spectrogram()
+
+    def forward(self, item):
+        waveform = item[0]
+        spectrogram = self.spectrogram_extractor(waveform)
+        return (spectrogram,) + item[1:]
+
+pack_to_hdf(dataset, packed_root, MyTransform())
+```
+
+Then you can load the pre-computed dataset using `HDFDataset`:
+```python
+from torchwrench.extras.hdf import HDFDataset
+
+packed_root = "path/to/packed_dataset.hdf"
+packed_dataset = HDFDataset(packed_root)
+packed_dataset[0]  # == first transformed item, i.e. transform(dataset[0])
+```
+
+## Contact
+Maintainer:
+- [Étienne Labbé](https://labbeti.github.io/) "Labbeti": labbeti.pub@gmail.com

torchwrench-0.7.0/README.md

@@ -0,0 +1,225 @@
+# torchwrench
+
+<center>
+
+<a href="https://www.python.org/">
+    <img alt="Python" src="https://img.shields.io/badge/-Python 3.9+-blue?style=for-the-badge&logo=python&logoColor=white">
+</a>
+<a href="https://github.com/Labbeti/torchwrench/actions">
+    <img alt="Build" src="https://img.shields.io/github/actions/workflow/status/Labbeti/torchwrench/test.yaml?branch=main&style=for-the-badge&logo=github">
+</a>
+<a href='https://torchwrench.readthedocs.io/'>
+    <img src='https://readthedocs.org/projects/torchwrench/badge/?version=stable&style=for-the-badge' alt='Documentation Status' />
+</a>
+<a href="https://pytorch.org/get-started/locally/">
+    <img alt="PyTorch" src="https://img.shields.io/badge/-PyTorch 1.10+-ee4c2c?style=for-the-badge&logo=pytorch&logoColor=white">
+</a>
+
+Collection of functions and modules to help development in PyTorch.
+
+</center>
+
+
+## Installation
+
+With pip:
+```bash
+pip install torchwrench
+```
+
+With uv:
+```bash
+uv add torchwrench
+```
+
+The main requirement is **[PyTorch](https://pytorch.org/)**.
+
+To check if the package is installed and show the package version, you can use the following command in your terminal:
+```bash
+torchwrench-info
+```
+
+This library works on all Python versions **>=3.9**, all PyTorch versions **>= 1.10**, and on **Linux, Mac and Windows** systems.
+
+## Examples
+
+`torchwrench` functions and modules can be used like `torch` ones. The default acronym for `torchwrench` is `tw`.
+
+### Label conversions
+Supports **multiclass** labels conversions between probabilities, classes indices, classes names and onehot encoding.
+
+```python
+import torchwrench as tw
+
+probs = tw.as_tensor([[0.9, 0.1], [0.4, 0.6]])
+names = tw.probs_to_name(probs, idx_to_name={0: "Cat", 1: "Dog"})
+# ["Cat", "Dog"]
+```
+
+This package also supports **multilabel** labels conversions between probabilities, classes multi-indices, classes multi-names and multihot encoding.
+
+```python
+import torchwrench as tw
+
+multihot = tw.as_tensor([[1, 0, 0], [0, 1, 1], [0, 0, 0]])
+indices = tw.multihot_to_indices(multihot)
+# [[0], [1, 2], []]
+```
+
+Finally, this packages includes the **powerset multilabel** conversions :
+
+```python
+import torchwrench as tw
+
+multihot = tw.as_tensor([[1, 0, 0], [0, 1, 1], [0, 0, 0]])
+indices = tw.multilabel_to_powerset(multihot, num_classes=3, max_set_size=2)
+# tensor([[0, 1, 0, 0, 0, 0, 0],
+#         [0, 0, 0, 0, 0, 0, 1],
+#         [1, 0, 0, 0, 0, 0, 0]])
+```
+
+### Typing
+
+Typing with number of dimensions :
+
+```python
+import torchwrench as tw
+
+x1 = tw.as_tensor([1, 2])
+print(isinstance(x1, tw.Tensor2D))  # False
+x2 = tw.as_tensor([[1, 2], [3, 4]])
+print(isinstance(x2, tw.Tensor2D))  # True
+```
+
+Typing with tensor dtype :
+
+```python
+import torchwrench as tw
+
+x1 = tw.as_tensor([1, 2], dtype=tw.int)
+print(isinstance(x1, tw.SignedIntegerTensor))  # True
+
+x2 = tw.as_tensor([1, 2], dtype=tw.long)
+print(isinstance(x2, tw.SignedIntegerTensor1D))  # True
+
+x3 = tw.as_tensor([1, 2], dtype=tw.float)
+print(isinstance(x3, tw.SignedIntegerTensor))  # False
+```
+
+### Padding & cropping
+
+Pad a specific dimension :
+
+```python
+import torchwrench as tw
+
+x = tw.rand(10, 3, 1)
+padded = tw.pad_dim(x, target_length=5, dim=1, pad_value=-1)
+# x2 has shape (10, 5, 1), padded with -1
+```
+
+Pad nested list of tensors to a single one :
+
+```python
+import torchwrench as tw
+
+tensors = [tw.rand(10, 2), [tw.rand(3)] * 5, tw.rand(0, 5)]
+padded = tw.pad_and_stack_rec(tensors, pad_value=0)
+# padded has shape (3, 10, 5), padded with 0
+```
+
+Remove values at a specific dimension :
+
+```python
+import torchwrench as tw
+
+x = tw.rand(10, 5, 3)
+cropped = tw.crop_dim(x, dim=1, target_length=2)
+# cropped has shape (10, 2, 3)
+```
+
+### Masking
+
+```python
+import torchwrench as tw
+
+x = tw.as_tensor([3, 1, 2])
+mask = tw.lengths_to_non_pad_mask(x, max_len=4)
+# Each row i contains x[i] True values for non-padding mask
+# tensor([[True, True, True, False],
+#         [True, False, False, False],
+#         [True, True, False, False]])
+```
+
+```python
+import torchwrench as tw
+
+x = tw.as_tensor([1, 2, 3, 4])
+mask = tw.as_tensor([True, True, False, False])
+result = tw.masked_mean(x, mask)
+# result contains the mean of the values marked as True: 1.5
+```
+
+### Others tensors manipulations!
+
+```python
+import torchwrench as tw
+
+x = tw.as_tensor([1, 2, 3, 4])
+result = tw.insert_at_indices(x, indices=[0, 2], values=5)
+# result contains tensor with inserted values: tensor([5, 1, 2, 5, 3, 4])
+```
+
+```python
+import torchwrench as tw
+
+perm = tw.randperm(10)
+inv_perm = tw.get_inverse_perm(perm)
+
+x1 = tw.rand(10)
+x2 = x1[perm]
+x3 = x2[inv_perm]
+# inv_perm are indices that allow us to get x3 from x2, i.e. x1 == x3 here
+```
+
+### Extra: pre-compute datasets to HDF files
+
+Here is an example of pre-computing spectrograms of torchaudio `SPEECHCOMMANDS` dataset, using `pack_dataset` function:
+
+```python
+from torchaudio.datasets import SPEECHCOMMANDS
+from torchaudio.transforms import Spectrogram
+from torchwrench import nn
+from torchwrench.extras.hdf import pack_to_hdf
+
+speech_commands_root = "path/to/speech_commands"
+packed_root = "path/to/packed_dataset.hdf"
+
+dataset = SPEECHCOMMANDS(speech_commands_root, download=True, subset="validation")
+# dataset[0] is a tuple, contains waveform and other metadata
+
+class MyTransform(nn.Module):
+    def __init__(self) -> None:
+        super().__init__()
+        self.spectrogram_extractor = Spectrogram()
+
+    def forward(self, item):
+        waveform = item[0]
+        spectrogram = self.spectrogram_extractor(waveform)
+        return (spectrogram,) + item[1:]
+
+pack_to_hdf(dataset, packed_root, MyTransform())
+```
+
+Then you can load the pre-computed dataset using `HDFDataset`:
+```python
+from torchwrench.extras.hdf import HDFDataset
+
+packed_root = "path/to/packed_dataset.hdf"
+packed_dataset = HDFDataset(packed_root)
+packed_dataset[0]  # == first transformed item, i.e. transform(dataset[0])
+```
+
+## Contact
+Maintainer:
+- [Étienne Labbé](https://labbeti.github.io/) "Labbeti": labbeti.pub@gmail.com

torchwrench-0.7.0/pyproject.toml

@@ -0,0 +1,104 @@
+[project]
+name = "torchwrench"
+authors = [
+    {name = "Étienne Labbé (Labbeti)", email = "labbeti.pub@gmail.com"},
+]
+description = "Collection of functions and modules to help development in PyTorch."
+readme = "README.md"
+requires-python = ">=3.9"
+keywords = ["pytorch", "deep-learning"]
+license = {file = "LICENSE"}
+classifiers = [
+    "Intended Audience :: Developers",
+    "Intended Audience :: Science/Research",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Topic :: Scientific/Engineering",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    "Operating System :: OS Independent",
+]
+maintainers = [
+    {name = "Étienne Labbé (Labbeti)", email = "labbeti.pub@gmail.com"},
+]
+dynamic = ["version"]
+dependencies = [
+    "pythonwrench~=0.1.0",
+    "torch>=1.10.0",
+    "typing-extensions>=4.10.0",
+]
+
+[project.urls]
+Homepage = "https://pypi.org/project/torchwrench/"
+Documentation = "https://torchwrench.readthedocs.io/"
+Repository = "https://github.com/Labbeti/torchwrench.git"
+Changelog = "https://github.com/Labbeti/torchwrench/blob/main/CHANGELOG.md"
+Tracker = "https://github.com/Labbeti/torchwrench/issues"
+
+[project.scripts]
+torchwrench-info = "torchwrench.entries:print_install_info"
+
+[build-system]
+requires = ["setuptools >= 61.0"]
+build-backend = "setuptools.build_meta"
+
+[tool.setuptools.packages.find]
+where = ["src"]  # list of folders that contain the packages (["."] by default)
+# package names should match these glob patterns (["*"] by default)
+# note: star after package names is required to work
+include = ["torchwrench*"]
+
+[tool.setuptools.dynamic]
+version = {attr = "torchwrench.__version__"}
+
+[tool.pyright]
+include = ["src"]
+exclude = ["**/__pycache__"]
+pythonVersion = "3.9"
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+
+[tool.coverage.run]
+source = ["src"]
+
+[tool.ruff.lint.extend-per-file-ignores]
+"__init__.py" = ["F401", "F403"]
+
+[tool.uv.sources]
+torchwrench = { workspace = true }
+
+[project.optional-dependencies]
+dev = [
+    "coverage[toml]>=7.9.1",
+    "flake8>=7.2.0",
+    "ipython>=8.18.1",
+    "pre-commit>=4.2.0",
+    "pytest>=8.4.1",
+    "ruff>=0.12.0",
+    "setuptools>=80.9.0",
+    "soundfile>=0.13.1",
+    "sphinx>=7.4.7",
+    "sphinx-press-theme>=0.8.0",
+    "twine>=6.1.0",
+    "typing-extensions>=4.10.0",
+    "ipykernel>=6.29.5",
+    "numpy!=2.0.0,!=2.0.1,!=2.0.2",
+]
+extras = [
+    "colorlog>=6.9.0",
+    "h5py>=3.14.0",
+    "numpy!=2.0.0,!=2.0.1,!=2.0.2",
+    "omegaconf>=2.3.0",
+    "pandas>=2.3.0",
+    "pyyaml>=6.0.2",
+    "safetensors>=0.5.3",
+    "scipy>=1.13.1",
+    "tensorboard>=2.19.0",
+    "torchaudio>=0.10.0",
+    "torchvision>=0.11.0",
+    "tqdm>=4.67.1",
+]

torchwrench-0.7.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

torchwrench-0.7.0/setup.py

@@ -0,0 +1,7 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+from setuptools import setup
+
+if __name__ == "__main__":
+    setup()