diff options
Diffstat (limited to 'candle-pyo3/tests')
| -rw-r--r-- | candle-pyo3/tests/bindings/test_module.py | 4 | ||||
| -rw-r--r-- | candle-pyo3/tests/native/test_tensor.py | 6 |
2 files changed, 5 insertions, 5 deletions
diff --git a/candle-pyo3/tests/bindings/test_module.py b/candle-pyo3/tests/bindings/test_module.py index 819dae5b..8d7a670d 100644 --- a/candle-pyo3/tests/bindings/test_module.py +++ b/candle-pyo3/tests/bindings/test_module.py @@ -74,7 +74,7 @@ def test_module_can_load_statedict(): a.load_state_dict(statedict) -def test_module_throws_on_shape_missmatch(): +def test_module_throws_on_shape_mismatch(): class A(Module): def __init__(self): super().__init__() @@ -121,7 +121,7 @@ def test_module_can_load_quantized_tensors(): assert a.t.ggml_dtype == "Q4_0" -def test_module_dequantizes_tensors_automaticaly(): +def test_module_dequantizes_tensors_automatically(): class A(Module): def __init__(self): super().__init__() diff --git a/candle-pyo3/tests/native/test_tensor.py b/candle-pyo3/tests/native/test_tensor.py index ef44fc4c..04eaca0d 100644 --- a/candle-pyo3/tests/native/test_tensor.py +++ b/candle-pyo3/tests/native/test_tensor.py @@ -84,7 +84,7 @@ def assert_bool(t: Tensor, expected: bool): assert bool(t.values()) == expected -def test_tensor_supports_equality_opperations_with_scalars(): +def test_tensor_supports_equality_operations_with_scalars(): t = Tensor(42.0) assert_bool(t == 42.0, True) @@ -106,7 +106,7 @@ def test_tensor_supports_equality_opperations_with_scalars(): assert_bool(t <= 42.0, True) -def test_tensor_supports_equality_opperations_with_tensors(): +def test_tensor_supports_equality_operations_with_tensors(): t = Tensor(42.0) same = Tensor(42.0) other = Tensor(43.0) @@ -130,7 +130,7 @@ def test_tensor_supports_equality_opperations_with_tensors(): assert_bool(t <= other, True) -def test_tensor_equality_opperations_can_broadcast(): +def test_tensor_equality_operations_can_broadcast(): # Create a decoder attention mask as a test case # e.g. # [[1,0,0] |