Skip to content

Commit 11661b9

Browse files
authored
Update dimensionality_reduction.py
1 parent 8d63f1b commit 11661b9

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

machine_learning/dimensionality_reduction.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010

1111
import doctest
1212
import logging
13+
1314
import numpy as np
1415
import pytest
1516
from scipy.linalg import eigh
@@ -417,7 +418,9 @@ def test_principal_component_analysis() -> None:
417418
"""Test function for Principal Component Analysis."""
418419
features = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
419420
dimensions = 2
420-
expected_output = np.array([[6.92820323, 8.66025404, 10.39230485], [3.0, 3.0, 3.0]])
421+
expected_output = np.array(
422+
[[6.92820323, 8.66025404, 10.39230485], [3.0, 3.0, 3.0]]
423+
)
421424

422425
output = principal_component_analysis(features, dimensions)
423426
if not np.allclose(expected_output, output):

0 commit comments

Comments
 (0)