Address comment

uxlfoundation · Sep 3, 2024 · b751aa2 · b751aa2
1 parent 3e51801
commit b751aa2
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/sklearnex/spmd/decomposition/tests/test_incremental_pca_spmd.py b/sklearnex/spmd/decomposition/tests/test_incremental_pca_spmd.py
@@ -57,8 +57,9 @@ def test_incremental_pca_fit_spmd_gold(dataframe, queue, whiten, dtype):
             [5.0, 32.0],
             [6.0, 64.0],
             [7.0, 128.0],
-        ]
-    ).astype(dtype=dtype)
+        ],
+        dtype=dtype,
+    )
     dpt_X = _convert_to_dataframe(X, sycl_queue=queue, target_df=dataframe)
     local_X = _get_local_tensor(X)
     local_dpt_X = _convert_to_dataframe(local_X, sycl_queue=queue, target_df=dataframe)
@@ -118,8 +119,9 @@ def test_incremental_pca_partial_fit_spmd_gold(
             [13.0, 32.0],
             [14.0, 64.0],
             [15.0, 128.0],
-        ]
-    ).astype(dtype=dtype)
+        ],
+        dtype=dtype,
+    )
     X_split = np.array_split(X, num_blocks)
     local_X = _get_local_tensor(X)
     split_local_X = np.array_split(local_X, num_blocks)