File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -31,8 +31,8 @@ def test_base_estimator_repr():
31
31
assert estimator .__repr__ () == "LinearRegression()"
32
32
33
33
# TODO(b/340891292): fix type error
34
- estimator = bigframes .ml .decomposition .PCA (n_components = 7 ) # type: ignore
35
- assert estimator .__repr__ () == "PCA(n_components=7)"
34
+ pca_estimator = bigframes .ml .decomposition .PCA (n_components = 7 )
35
+ assert pca_estimator .__repr__ () == "PCA(n_components=7)"
36
36
37
37
38
38
@pytest .mark .skipif (sklearn_linear_model is None , reason = "requires sklearn" )
@@ -50,6 +50,6 @@ def test_base_estimator_repr_matches_sklearn():
50
50
assert estimator .__repr__ () == sklearn_estimator .__repr__ ()
51
51
52
52
# TODO(b/340891292): fix type error
53
- estimator = bigframes .ml .decomposition .PCA (n_components = 7 ) # type: ignore
53
+ pca_estimator = bigframes .ml .decomposition .PCA (n_components = 7 )
54
54
sklearn_estimator = sklearn_decomposition .PCA (n_components = 7 )
55
- assert estimator .__repr__ () == sklearn_estimator .__repr__ ()
55
+ assert pca_estimator .__repr__ () == sklearn_estimator .__repr__ ()
You can’t perform that action at this time.
0 commit comments