Skip to content

Commit a4a6439

Browse files
Merge branch 'main' into helmeleegy-SNOW-1819523
2 parents 38c3276 + 051d837 commit a4a6439

File tree

1 file changed

+6
-8
lines changed

1 file changed

+6
-8
lines changed

src/snowflake/snowpark/modin/plugin/_internal/apply_utils.py

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -842,15 +842,13 @@ def convert_numpy_int_result_to_int(value: Any) -> Any:
842842

843843

844844
DUMMY_BOOL_INPUT = native_pd.Series([False, True])
845-
DUMMY_INT_INPUT = native_pd.Series(
846-
[-37, -9, -2, -1, 0, 2, 3, 5, 7, 9, 13, 16, 20]
847-
+ np.power(10, np.arange(19)).tolist()
848-
+ np.multiply(-1, np.power(10, np.arange(19))).tolist()
849-
)
845+
# Note: we use only small dummy values here to avoid the risk of certain callables
846+
# taking a long time to execute (where execution time is a function of the input value).
847+
# As a downside this reduces diversity in input data so will reduce the effectiveness
848+
# type inference framework in some rare cases.
849+
DUMMY_INT_INPUT = native_pd.Series([-37, -9, -2, -1, 0, 2, 3, 5, 7, 9, 13, 16, 20, 101])
850850
DUMMY_FLOAT_INPUT = native_pd.Series(
851-
[-9.9, -2.2, -1.0, 0.0, 0.5, 0.33, None, 0.99, 2.0, 3.0, 5.0, 7.7, 9.898989]
852-
+ np.power(10.1, np.arange(19)).tolist()
853-
+ np.multiply(-1.0, np.power(10.1, np.arange(19))).tolist()
851+
[-9.9, -2.2, -1.0, 0.0, 0.5, 0.33, None, 0.99, 2.0, 3.0, 5.0, 7.7, 9.898989, 100.1]
854852
)
855853
DUMMY_STRING_INPUT = native_pd.Series(
856854
["", "a", "A", "0", "1", "01", "123", "-1", "-12", "true", "True", "false", "False"]

0 commit comments

Comments
 (0)