Changae california_housing API

Author: PicoCentauri

tests/test_sample_simple_cur.py

@@ -1,15 +1,16 @@
 import unittest
 
 import numpy as np
-from sklearn.datasets import fetch_california_housing as load
+from sklearn.datasets import fetch_california_housing
 
 from skmatter.sample_selection import CUR, FPS
 
 
 class TestCUR(unittest.TestCase):
     def setUp(self):
-        self.X, _ = load(return_X_y=True)
-        self.X = self.X[FPS(n_to_select=100).fit(self.X).selected_idx_]
+        housing = fetch_california_housing()
+        X = housing.data
+        self.X = X[FPS(n_to_select=100).fit(X).selected_idx_]
         self.n_select = min(20, min(self.X.shape) // 2)
 
     def test_sample_transform(self):