resolve the comments

Author: jialuoo

bigframes/functions/_utils.py

@@ -81,7 +81,7 @@ def _get_updated_package_requirements(
             msg = bfe.format_message(
                 "Numpy, Pandas, and Pyarrow version may not precisely match your local environment."
             )
-            warnings.warn(msg, category=bfe.PreviewWarning)
+            warnings.warn(msg, category=bfe.FunctionPackageVersionWarning)
             requirements.append("pandas")
             requirements.append("pyarrow")
             requirements.append("numpy")
@@ -94,6 +94,8 @@ def _get_updated_package_requirements(
             requirements.append(f"pyarrow=={pyarrow.__version__}")
             requirements.append(f"numpy=={numpy.__version__}")
 
+    # TODO(b/435023957): Fix the issue of potential duplicate package versions
+    # when `package_requirements` also contains `pandas/pyarrow/numpy`.
     if package_requirements:
         requirements.extend(package_requirements)
 

tests/system/large/functions/test_managed_function.py

@@ -714,9 +714,9 @@ def test_managed_function_df_apply_axis_1_aggregates(session, dataset_id, scalar
     try:
 
         def analyze(row):
-            # TODO(b/435021126): Remove explicit type conversion of the field
-            # "name" after the issue has been addressed. It is added only to
-            # accept partial pandas parity for the time being.
+            # TODO(b/435021126): Remove explicit type conversion of the fields
+            # after the issue has been addressed. It is added only to accept
+            # partial pandas parity for the time being.
             return str(
                 {
                     "dtype": row.dtype,
@@ -730,7 +730,7 @@ def analyze(row):
             )
 
         with pytest.warns(
-            bfe.PreviewWarning,
+            bfe.FunctionPackageVersionWarning,
             match=("Numpy, Pandas, and Pyarrow version may not precisely match."),
         ):