Skip tests properly (#31308)
* Skip tests properly * [test_all] * Add 'reason' as kwarg for skipTest * [test_all] Fix up * [test_all]
This commit is contained in:
@@ -332,20 +332,23 @@ class QuantoQuantizationOffloadTest(QuantoQuantizationTest):
|
||||
"lm_head": 0,
|
||||
}
|
||||
|
||||
# the execution device is a gpu
|
||||
@unittest.skip(reason="The execution device is a gpu")
|
||||
def test_generate_quality_cpu(self):
|
||||
pass
|
||||
|
||||
# we can't save offloaded values
|
||||
@unittest.skip(reason="We can't save offloaded values")
|
||||
def test_serialization_bin(self):
|
||||
pass
|
||||
|
||||
@unittest.skip
|
||||
def test_serialization_safetensors(self):
|
||||
pass
|
||||
|
||||
@unittest.skip
|
||||
def test_compare_with_quanto(self):
|
||||
pass
|
||||
|
||||
@unittest.skip
|
||||
def test_load_from_quanto_saved(self):
|
||||
pass
|
||||
|
||||
@@ -370,7 +373,7 @@ class QuantoQuantizationOffloadTest(QuantoQuantizationTest):
|
||||
)
|
||||
|
||||
|
||||
@unittest.skip("Skipping test class because serialization is not supported yet")
|
||||
@unittest.skip(reason="Skipping test class because serialization is not supported yet")
|
||||
class QuantoQuantizationSerializationTest(QuantoQuantizationTest):
|
||||
"""
|
||||
Perform the same tests as in QuantoQuantizationTest but with a serialized model.
|
||||
@@ -403,7 +406,7 @@ class QuantoQuantizationSerializationTest(QuantoQuantizationTest):
|
||||
)
|
||||
|
||||
|
||||
@unittest.skip("Skipping test class because serialization is not supported yet")
|
||||
@unittest.skip(reason="Skipping test class because serialization is not supported yet")
|
||||
class QuantoQuantizationSerializationCudaTest(QuantoQuantizationTest):
|
||||
"""
|
||||
Perform the same tests as in QuantoQuantizationTest but with model on cuda
|
||||
@@ -422,7 +425,7 @@ class QuantoQuantizationQBitsTensorOffloadTest(QuantoQuantizationOffloadTest):
|
||||
weights = "int4"
|
||||
|
||||
|
||||
@unittest.skip("Skipping test class because serialization is not supported yet")
|
||||
@unittest.skip(reason="Skipping test class because serialization is not supported yet")
|
||||
class QuantoQuantizationQBitsTensorSerializationTest(QuantoQuantizationSerializationTest):
|
||||
EXPECTED_OUTPUTS = "Hello my name is Nils, I am a student of the University"
|
||||
weights = "int4"
|
||||
|
||||
Reference in New Issue
Block a user