fix errors

Author: topper-123

doc/source/user_guide/basics.rst

@@ -1119,7 +1119,7 @@ resulting column names will be the transforming functions.
 
 .. ipython:: python
 
-   tsdf["A"].transform([np.abs, lambda x: x + 1])
+   tsdf["A"].transform([np.abs, lambda x: x + 1], series_ops_only=True)
 
 
 Transforming with a dict

pandas/tests/apply/common.py

@@ -1,3 +1,5 @@
+import numpy as np
+
 import pandas as pd
 import pandas._testing as tm
 from pandas.api.types import is_list_like
@@ -12,7 +14,9 @@
 
 def transform_obj(obj, func, *args, axis=0, series_ops_only=False, **kwargs):
     """helper function to ease use of series_ops_only and deprecation warning."""
-    if series_ops_only:
+    if isinstance(func, np.ufunc):
+        result = obj.transform(func, axis, *args, **kwargs)
+    elif series_ops_only:
         result = obj.transform(
             func, axis, *args, series_ops_only=series_ops_only, **kwargs
         )

pandas/tests/apply/test_frame_transform.py

@@ -28,15 +28,15 @@ def unpack_obj(obj, klass, axis):
     return obj
 
 
-def test_transform_ufunc(axis, float_frame, frame_or_series, series_ops_only):
+def test_transform_ufunc(axis, float_frame, frame_or_series):
     # GH 35964
     obj = unpack_obj(float_frame, frame_or_series, axis)
 
     with np.errstate(all="ignore"):
         f_sqrt = np.sqrt(obj)
 
     # ufunc
-    result = transform_obj(obj, np.sqrt, axis=axis, series_ops_only=series_ops_only)
+    result = transform_obj(obj, np.sqrt, axis=axis)
     expected = f_sqrt
     tm.assert_equal(result, expected)