=================================== FAILURES ===================================
_________ test_llm_gemini_pro_score_params[gemini-2.0-flash-lite-001] __________
[gw8] linux -- Python 3.9.20 /tmpfs/src/github/python-bigquery-dataframes/.nox/system-3-9/bin/python

llm_fine_tune_df_default_index =   label                                               text  row_number  \
0     4                i answered feeling ra...iment analysis on the following ...  
9  Please do sentiment analysis on the following ...  
...

[30 rows x 4 columns]
model_name = 'gemini-2.0-flash-lite-001'

    @pytest.mark.parametrize(
        "model_name",
        (
            "gemini-2.0-flash-001",
            "gemini-2.0-flash-lite-001",
        ),
    )
    def test_llm_gemini_pro_score_params(llm_fine_tune_df_default_index, model_name):
        model = llm.GeminiTextGenerator(model_name=model_name)
    
        # Check score to ensure the model was fitted
>       score_result = model.score(
            X=llm_fine_tune_df_default_index["prompt"],
            y=llm_fine_tune_df_default_index["label"],
            task_type="classification",
        ).to_pandas()

tests/system/small/ml/test_llm.py:785: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
bigframes/core/log_adapter.py:182: in wrapper
    return method(*args, **kwargs)
bigframes/ml/llm.py:829: in score
    return self._bqml_model.llm_evaluate(input_data, task_type)
bigframes/ml/core.py:308: in llm_evaluate
    return self._session.read_gbq_query(sql, allow_large_results=True)
bigframes/core/log_adapter.py:182: in wrapper
    return method(*args, **kwargs)
bigframes/session/__init__.py:716: in read_gbq_query
    return self._loader.read_gbq_query(  # type: ignore # for dry_run overload
bigframes/session/loader.py:1072: in read_gbq_query
    destination, query_job = self._query_to_destination(
bigframes/session/loader.py:1236: in _query_to_destination
    query_job = self._start_query_with_job(
bigframes/session/loader.py:1303: in _start_query_with_job
    _, query_job = bf_io_bigquery.start_query_with_client(
bigframes/session/_io/bigquery/__init__.py:390: in start_query_with_client
    results_iterator = query_job.result()
.nox/system-3-9/lib/python3.9/site-packages/google/cloud/bigquery/job/query.py:1706: in result
    while not is_job_done():
.nox/system-3-9/lib/python3.9/site-packages/google/api_core/retry/retry_unary.py:294: in retry_wrapped_func
    return retry_target(
.nox/system-3-9/lib/python3.9/site-packages/google/api_core/retry/retry_unary.py:147: in retry_target
    result = target()
.nox/system-3-9/lib/python3.9/site-packages/google/cloud/bigquery/job/query.py:1675: in is_job_done
    self._reload_query_results(retry=retry, **reload_query_results_kwargs)
.nox/system-3-9/lib/python3.9/site-packages/google/cloud/bigquery/job/query.py:1467: in _reload_query_results
    self._query_results = self._client._get_query_results(
.nox/system-3-9/lib/python3.9/site-packages/google/cloud/bigquery/client.py:2113: in _get_query_results
    resource = self._call_api(
.nox/system-3-9/lib/python3.9/site-packages/google/cloud/bigquery/client.py:861: in _call_api
    return call()
.nox/system-3-9/lib/python3.9/site-packages/google/api_core/retry/retry_unary.py:294: in retry_wrapped_func
    return retry_target(
.nox/system-3-9/lib/python3.9/site-packages/google/api_core/retry/retry_unary.py:147: in retry_target
    result = target()
.nox/system-3-9/lib/python3.9/site-packages/google/cloud/_http/__init__.py:482: in api_request
    response = self._make_request(
.nox/system-3-9/lib/python3.9/site-packages/google/cloud/_http/__init__.py:341: in _make_request
    return self._do_request(
.nox/system-3-9/lib/python3.9/site-packages/google/cloud/_http/__init__.py:379: in _do_request
    return self.http.request(
.nox/system-3-9/lib/python3.9/site-packages/google/auth/transport/requests.py:549: in request
    response = super(AuthorizedSession, self).request(
.nox/system-3-9/lib/python3.9/site-packages/requests/sessions.py:529: in request
    resp = self.send(prep, **send_kwargs)
.nox/system-3-9/lib/python3.9/site-packages/requests/sessions.py:645: in send
    r = adapter.send(request, **kwargs)
.nox/system-3-9/lib/python3.9/site-packages/requests/adapters.py:440: in send
    resp = conn.urlopen(
.nox/system-3-9/lib/python3.9/site-packages/urllib3/connectionpool.py:716: in urlopen
    httplib_response = self._make_request(
.nox/system-3-9/lib/python3.9/site-packages/urllib3/connectionpool.py:468: in _make_request
    six.raise_from(e, None)
<string>:3: in raise_from
    ???
.nox/system-3-9/lib/python3.9/site-packages/urllib3/connectionpool.py:463: in _make_request
    httplib_response = conn.getresponse()
/usr/local/lib/python3.9/http/client.py:1377: in getresponse
    response.begin()
/usr/local/lib/python3.9/http/client.py:320: in begin
    version, status, reason = self._read_status()
/usr/local/lib/python3.9/http/client.py:281: in _read_status
    line = str(self.fp.readline(_MAXLINE + 1), "iso-8859-1")
/usr/local/lib/python3.9/socket.py:716: in readinto
    return self._sock.recv_into(b)
/usr/local/lib/python3.9/ssl.py:1275: in recv_into
    return self.read(nbytes, buffer)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 

self = <ssl.SSLSocket [closed] fd=-1, family=AddressFamily.AF_INET, type=SocketKind.SOCK_STREAM, proto=6>
len = 8192, buffer = <memory at 0x1539fab08700>

    def read(self, len=1024, buffer=None):
        """Read up to LEN bytes and return them.
        Return zero-length string on EOF."""
    
        self._checkClosed()
        if self._sslobj is None:
            raise ValueError("Read on closed or unwrapped SSL socket.")
        try:
            if buffer is not None:
>               return self._sslobj.read(len, buffer)
E               Failed: Timeout (>900.0s) from pytest-timeout.

/usr/local/lib/python3.9/ssl.py:1133: Failed
----------------------------- Captured stdout call -----------------------------
+++++++++++++++++++++++++++++++++++ Timeout ++++++++++++++++++++++++++++++++++++