Add torchtune to docker image packages (#1456)

Kaggle · Dec 16, 2024 · 2cb0a25 · 2cb0a25
1 parent 18a67b0
commit 2cb0a25
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/kaggle_requirements.txt b/kaggle_requirements.txt
@@ -140,6 +140,7 @@ tensorflow_decision_forests
 timm
 torchinfo
 torchmetrics
+torchtune
 tsfresh
 vtk
 wandb

diff --git a/tests/test_torchtune.py b/tests/test_torchtune.py
@@ -0,0 +1,9 @@
+import unittest
+
+import subprocess
+
+class TestTorchtune(unittest.TestCase):
+    def test_help(self):
+        ret_code = subprocess.run(["tune", "--help"])
+        self.assertEqual(0, ret_code.returncode)
+        self.assertIsNone(ret_code.stderr)
-Original file line number
+Diff line change
@@ Expand Up / @@ -140,6 +140,7 @@ tensorflow_decision_forests @@
     timm
     torchinfo
     torchmetrics
+    torchtune
     tsfresh
     vtk
     wandb
@@ Expand Down @@