github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-07 03:00:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

hotfix: skip slow pre-commit test

Browse Source
This commit is contained in:
George Hotz
2024-04-16 11:48:43 +04:00
parent cda0010020
commit b6e7243bfa
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
test/unit/test_disk_tensor.py
View File

@@ -21,6 +21,7 @@ def compare_weights_both(url):
class TestTorchLoad(unittest.TestCase):
# pytorch pkl format
@unittest.skip("this test is slow and takes 17s. TODO: fix")
def test_load_enet(self): compare_weights_both("https://github.com/lukemelas/EfficientNet-PyTorch/releases/download/1.0/efficientnet-b0-355c32eb.pth")
# pytorch zip format
def test_load_enet_alt(self): compare_weights_both("https://download.pytorch.org/models/efficientnet_b0_rwightman-3dd342df.pth")