numpy fix for slow test

Roja Reddy Sareddy · Roja Reddy Sareddy · commit 2b2c190abd71 · 2025-10-09T19:32:01.000-07:00
diff --git a/tests/integ/sagemaker/feature_store/feature_processor/test_feature_processor_integ.py b/tests/integ/sagemaker/feature_store/feature_processor/test_feature_processor_integ.py
@@ -231,7 +231,7 @@ def transform(raw_s3_data_as_df):
         )
 
 
-#@pytest.mark.slow_test
+@pytest.mark.slow_test
 def test_feature_processor_transform_with_customized_data_source(
     sagemaker_session,
 ):
@@ -373,7 +373,7 @@ def transform(raw_s3_data_as_df):
         )
 
 
-@pytest.mark.slow_test
+# @pytest.mark.slow_test
 @pytest.mark.flaky(reruns=5, reruns_delay=2)
 def test_feature_processor_transform_offline_only_store_ingestion(
     sagemaker_session,

Original file line number	Diff line number	Diff line change
`@@ -231,7 +231,7 @@ def transform(raw_s3_data_as_df):`
`231`	`231`	`)`
`232`	`232`
`233`	`233`
`234`		`-#@pytest.mark.slow_test`
	`234`	`+@pytest.mark.slow_test`
`235`	`235`	`def test_feature_processor_transform_with_customized_data_source(`
`236`	`236`	`sagemaker_session,`
`237`	`237`	`):`
`@@ -373,7 +373,7 @@ def transform(raw_s3_data_as_df):`
`373`	`373`	`)`
`374`	`374`
`375`	`375`
`376`		`-@pytest.mark.slow_test`
	`376`	`+# @pytest.mark.slow_test`
`377`	`377`	`@pytest.mark.flaky(reruns=5, reruns_delay=2)`
`378`	`378`	`def test_feature_processor_transform_offline_only_store_ingestion(`
`379`	`379`	`sagemaker_session,`