diff --git a/kaggle_requirements.txt b/kaggle_requirements.txt index 30f819c5..8379d0f2 100644 --- 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 new file mode 100644 index 00000000..b2090ecc --- /dev/null +++ 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) \ No newline at end of file