Skip to content

Commit 2cb0a25

Browse files
authored
Add torchtune to docker image packages (#1456)
1 parent 18a67b0 commit 2cb0a25

File tree

2 files changed

+10
-0
lines changed

2 files changed

+10
-0
lines changed

kaggle_requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -140,6 +140,7 @@ tensorflow_decision_forests
140140
timm
141141
torchinfo
142142
torchmetrics
143+
torchtune
143144
tsfresh
144145
vtk
145146
wandb

tests/test_torchtune.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
import unittest
2+
3+
import subprocess
4+
5+
class TestTorchtune(unittest.TestCase):
6+
def test_help(self):
7+
ret_code = subprocess.run(["tune", "--help"])
8+
self.assertEqual(0, ret_code.returncode)
9+
self.assertIsNone(ret_code.stderr)

0 commit comments

Comments
 (0)