address comments

Author: sfc-gh-helmeleegy

src/snowflake/snowpark/modin/plugin/utils/numpy_to_pandas.py

@@ -243,7 +243,7 @@ def percentile_mapper(
     Numpy np.percentile signature:
     Return the q-th percentile(s) of an array, Series, or DataFrame (a).
 
-    Pandas pd.unique signature:
+    Pandas Series.quantile signature:
     Return the q-th quantile(s) of a Series or DataFrame.
 
     Parameters
@@ -278,7 +278,7 @@ def percentile_mapper(
     if interpolation is not None:
         return NotImplemented
     input_values = a
-    if isinstance(q, float):
+    if isinstance(q, (float, int)):
         q = q / 100
     else:
         q = [percentage / 100 for percentage in q]

tests/integ/modin/test_numpy.py

@@ -165,19 +165,19 @@ def test_np_percentile(q):
     # Verify that numpy throws type errors when we return NotImplemented
     # when using optional parameters
     with pytest.raises(TypeError):
-        np.percentile(y_true_snow_2d, axis=1)
+        np.percentile(y_true_snow_2d, 50, axis=1)
     with pytest.raises(TypeError):
-        np.percentile(y_true_snow_2d, out=np.zeros(1))
+        np.percentile(y_true_snow_2d, 50, out=np.zeros(1))
     with pytest.raises(TypeError):
-        np.percentile(y_true_snow_2d, overwrite_input=True)
+        np.percentile(y_true_snow_2d, 50, overwrite_input=True)
     with pytest.raises(TypeError):
-        np.percentile(y_true_snow_2d, method="inverted_cdf")
+        np.percentile(y_true_snow_2d, 50, method="inverted_cdf")
     with pytest.raises(TypeError):
-        np.percentile(y_true_snow_2d, keepdims=True)
+        np.percentile(y_true_snow_2d, 50, keepdims=True)
     with pytest.raises(TypeError):
-        np.percentile(y_true_snow_2d, weights=[0.25, 0.25, 0.25, 0.25])
+        np.percentile(y_true_snow_2d, 50, weights=[0.25, 0.25, 0.25, 0.25])
     with pytest.raises(TypeError):
-        np.percentile(y_true_snow_2d, interpolation="inverted_cdf")
+        np.percentile(y_true_snow_2d, 50, interpolation="inverted_cdf")
 
 
 def test_logical_operators():