github-starred/TinyTorch
2
0
Fork 0
mirror of https://github.com/MLSysBook/TinyTorch.git synced 2026-05-31 18:48:33 -05:00
Code Issues 7 Packages Projects Releases Wiki Activity

Fix test function calls in spatial and dataloader modules - move test calls outside __main__ blocks

Browse Source
This commit is contained in:
Vijay Janapa Reddi
2025-07-20 12:54:15 -04:00
parent 30b8bc3b59
commit 3bc71f6015
2 changed files with 18 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
modules/source/06_spatial/spatial_dev.py
View File

@@ -821,9 +821,17 @@ def test_module_conv2d_tensor_compatibility():
## 🤖 AUTO TESTING
"""
# %%
# Run the integration test
test_module_conv2d_tensor_compatibility()
# %% [markdown]
"""
## 🤖 AUTO TESTING
"""
# %%
if __name__ == "__main__":
test_module_conv2d_tensor_compatibility()
from tito.tools.testing import run_module_tests_auto
# Automatically discover and run all tests in this module

9
modules/source/08_dataloader/dataloader_dev.py
View File

@@ -1090,6 +1090,15 @@ def test_module_dataloader_tensor_yield():
## 🤖 AUTO TESTING
"""
# %%
# Run the integration test
test_module_dataloader_tensor_yield()
# %% [markdown]
"""
## 🤖 AUTO TESTING
"""
# %%
if __name__ == "__main__":
from tito.tools.testing import run_module_tests_auto