Skip to content

Commit 837bdcf

Browse files
author
Roja Reddy Sareddy
committed
Revert 'Add numpy 2.0 support'
1 parent a4d0e29 commit 837bdcf

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

tests/integ/sagemaker/feature_store/feature_processor/test_feature_processor_integ.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1108,15 +1108,15 @@ def get_expected_dataframe():
11081108
expected_dataframe = pd.read_csv(os.path.join(_FEATURE_PROCESSOR_DIR, "car-data.csv"))
11091109
expected_dataframe["Model"].replace("^\d\d\d\d\s", "", regex=True, inplace=True) # noqa: W605
11101110
expected_dataframe["Mileage"].replace("(,)|(mi\.)", "", regex=True, inplace=True) # noqa: W605
1111-
expected_dataframe["Mileage"].replace("Not available", np.nan, inplace=True)
1111+
expected_dataframe["Mileage"].replace("Not available", np.NaN, inplace=True)
11121112
expected_dataframe["Price"].replace("\$", "", regex=True, inplace=True) # noqa: W605
11131113
expected_dataframe["Price"].replace(",", "", regex=True, inplace=True)
11141114
expected_dataframe["MSRP"].replace(
11151115
"(^MSRP\s\\$)|(,)", "", regex=True, inplace=True # noqa: W605
11161116
)
1117-
expected_dataframe["MSRP"].replace("Not specified", np.nan, inplace=True)
1117+
expected_dataframe["MSRP"].replace("Not specified", np.NaN, inplace=True)
11181118
expected_dataframe["MSRP"].replace(
1119-
"\\$\d+[a-zA-Z\s]+", np.nan, regex=True, inplace=True # noqa: W605
1119+
"\\$\d+[a-zA-Z\s]+", np.NaN, regex=True, inplace=True # noqa: W605
11201120
)
11211121
expected_dataframe["Mileage"] = expected_dataframe["Mileage"].astype(float)
11221122
expected_dataframe["Price"] = expected_dataframe["Price"].astype(float)

tests/integ/sagemaker/remote_function/test_decorator.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@
2020
import logging
2121
import random
2222
import string
23-
import numpy as np
2423
import pandas as pd
2524
import subprocess
2625
import shlex

0 commit comments

Comments
 (0)