From 1795a5c881e10b161f058fd13a004493f72385cd Mon Sep 17 00:00:00 2001 From: Garrett Wu Date: Wed, 25 Sep 2024 23:47:59 +0000 Subject: [PATCH] test: re-enable remote model tests --- noxfile.py | 3 +-- tests/system/small/ml/test_core.py | 1 - tests/system/small/ml/test_remote.py | 2 -- 3 files changed, 1 insertion(+), 5 deletions(-) diff --git a/noxfile.py b/noxfile.py index c704da00a5..82e746a5c6 100644 --- a/noxfile.py +++ b/noxfile.py @@ -442,8 +442,7 @@ def cover(session): "--show-missing", "--include=tests/unit/*", "--include=tests/system/small/*", - # TODO(b/353775058) resume coverage to 100 when the issue is fixed. - "--fail-under=99", + "--fail-under=100", ) session.run("coverage", "erase") diff --git a/tests/system/small/ml/test_core.py b/tests/system/small/ml/test_core.py index 6b852e87af..5e21958dad 100644 --- a/tests/system/small/ml/test_core.py +++ b/tests/system/small/ml/test_core.py @@ -311,7 +311,6 @@ def test_model_detect_anomalies( ) -@pytest.mark.skip("b/353775058 BQML internal error") def test_remote_model_predict( bqml_linear_remote_model: core.BqmlModel, new_penguins_df ): diff --git a/tests/system/small/ml/test_remote.py b/tests/system/small/ml/test_remote.py index c52c452244..5036cdadfc 100644 --- a/tests/system/small/ml/test_remote.py +++ b/tests/system/small/ml/test_remote.py @@ -13,12 +13,10 @@ # limitations under the License. import pandas as pd -import pytest from bigframes.ml import remote -@pytest.mark.skip("b/353775058 BQML internal error") def test_remote_linear_vertex_model_predict( linear_remote_vertex_model: remote.VertexAIModel, new_penguins_df ):