asdfasdf

Author: calderjo

tests/test_torchtune.py

@@ -3,8 +3,14 @@
 
 class TestTorchtune(unittest.TestCase):
     def test_help(self):
-        result = subprocess.run(["tune", "--help"], stdout=subprocess.PIPE)
+        result = subprocess.run(
+            ["tune", "--help"], 
+            capture_output=True,
+            text=True
+        )
 
         self.assertEqual(0, result.returncode)
-        self.assertIsNone(result.stderr)
-        self.assertIn("Download a model from the Hugging Face Hub or Kaggle Model Hub.", result.stdout.decode("utf-8"))
+        self.assertIn(
+            "Download a model from the Hugging Face Hub or Kaggle", 
+            result.stdout
+        )