diff --git a/machine_learning/dimensionality_reduction.py b/machine_learning/dimensionality_reduction.py index 6080d0a4f11f..3912a9d9cf5e 100644 --- a/machine_learning/dimensionality_reduction.py +++ b/machine_learning/dimensionality_reduction.py @@ -143,7 +143,7 @@ def linear_discriminant_analysis( assert classes > dimensions # Check if features have been already loaded - if features.any: + if features.any(): _, eigenvectors = eigh( covariance_between_classes(features, labels, classes), covariance_within_classes(features, labels, classes), @@ -363,6 +363,11 @@ def test_linear_discriminant_analysis() -> None: ) assert error_info.type is AssertionError + with pytest.raises(AssertionError): + linear_discriminant_analysis( + np.array([]), np.array([]), classes=2, dimensions=1 + ) + def test_principal_component_analysis() -> None: features = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])