diff --git a/sdk/python/tests/integration/e2e/test_go_feature_server.py b/sdk/python/tests/integration/e2e/test_go_feature_server.py index bf6d463609..4fd003c194 100644 --- a/sdk/python/tests/integration/e2e/test_go_feature_server.py +++ b/sdk/python/tests/integration/e2e/test_go_feature_server.py @@ -122,7 +122,7 @@ def grpc_client(grpc_server_port): @pytest.mark.integration -# @pytest.mark.goserver Disabling because the go fs tests are flaking in CI. TODO(achals): uncomment after fixed. +@pytest.mark.goserver def test_go_grpc_server(grpc_client): resp: GetOnlineFeaturesResponse = grpc_client.GetOnlineFeatures( GetOnlineFeaturesRequest( @@ -148,7 +148,7 @@ def test_go_grpc_server(grpc_client): @pytest.mark.integration -# @pytest.mark.goserver Disabling because the go fs tests are flaking in CI. TODO(achals): uncomment after fixed. +@pytest.mark.goserver def test_go_http_server(http_server_port): response = requests.post( f"http://localhost:{http_server_port}/get-online-features", @@ -186,7 +186,7 @@ def test_go_http_server(http_server_port): @pytest.mark.integration -# @pytest.mark.goserver Disabling because the go fs tests are flaking in CI. TODO(achals): uncomment after fixed. +@pytest.mark.goserver @pytest.mark.universal_offline_stores @pytest.mark.parametrize("full_feature_names", [True, False], ids=lambda v: str(v)) def test_feature_logging( diff --git a/sdk/python/tests/integration/online_store/test_universal_online.py b/sdk/python/tests/integration/online_store/test_universal_online.py index c068e04111..9fdb19eae7 100644 --- a/sdk/python/tests/integration/online_store/test_universal_online.py +++ b/sdk/python/tests/integration/online_store/test_universal_online.py @@ -443,7 +443,7 @@ def test_online_retrieval_with_event_timestamps( @pytest.mark.integration @pytest.mark.universal_online_stores -# @pytest.mark.goserver Disabling because the go fs tests are flaking in CI. TODO(achals): uncomment after fixed. +@pytest.mark.goserver @pytest.mark.parametrize("full_feature_names", [True, False], ids=lambda v: str(v)) def test_stream_feature_view_online_retrieval( environment, universal_data_sources, feature_server_endpoint, full_feature_names