fix the tests

Author: jialuoo

bigframes/session/__init__.py

@@ -2065,7 +2065,7 @@ def read_gbq_function(
         parameter.
 
             >>> @bpd.remote_function(cloud_function_service_account="default")
-            ... def row_sum(s: bpd.Series) -> float:
+            ... def row_sum(s: pd.Series) -> float:
             ...     return s['a'] + s['b'] + s['c']
 
             >>> row_sum_ref = bpd.read_gbq_function(

tests/unit/functions/test_remote_function.py

@@ -17,25 +17,12 @@
 import pandas
 import pytest
 
+import bigframes.exceptions
 import bigframes.functions.function as bff
-import bigframes.series
 from bigframes.testing import mocks
 
 
-@pytest.mark.parametrize(
-    "series_type",
-    (
-        pytest.param(
-            pandas.Series,
-            id="pandas.Series",
-        ),
-        pytest.param(
-            bigframes.series.Series,
-            id="bigframes.series.Series",
-        ),
-    ),
-)
-def test_series_input_types_to_str(series_type):
+def test_series_input_types_to_str():
     """Check that is_row_processor=True uses str as the input type to serialize a row."""
     session = mocks.create_bigquery_session()
     remote_function_decorator = bff.remote_function(
@@ -48,7 +35,7 @@ def test_series_input_types_to_str(series_type):
     ):
 
         @remote_function_decorator
-        def axis_1_function(myparam: series_type) -> str:  # type: ignore
+        def axis_1_function(myparam: pandas.Series) -> str:  # type: ignore
             return "Hello, " + myparam["str_col"] + "!"  # type: ignore
 
     # Still works as a normal function.