Fix tests

tharapalanivel · tharapalanivel · commit 56238ab313b9 · 2025-01-07T10:31:39.000-08:00
Signed-off-by: Thara Palanivel &lt;130496890+tharapalanivel@users.noreply.github.com&gt;
diff --git a/tests/test_run_quant.py b/tests/test_run_quant.py
@@ -54,6 +54,9 @@
 )
 
 
+@pytest.mark.skipif(
+    not torch.cuda.is_available(), reason="Only runs if GPUs are available"
+)
 def test_run_train_requires_output_dir():
     """Check fails when output dir not provided."""
     updated_output_dir_opt_args = copy.deepcopy(OPT_ARGS)

Original file line number	Diff line number	Diff line change
`@@ -54,6 +54,9 @@`
`54`	`54`	`)`
`55`	`55`
`56`	`56`
	`57`	`+@pytest.mark.skipif(`
	`58`	`+ not torch.cuda.is_available(), reason="Only runs if GPUs are available"`
	`59`	`+)`
`57`	`60`	`def test_run_train_requires_output_dir():`
`58`	`61`	`"""Check fails when output dir not provided."""`
`59`	`62`	`updated_output_dir_opt_args = copy.deepcopy(OPT_ARGS)`