Skip to content

Commit 11cf040

Browse files
committed
fix black format issues and update test for sklearn
1 parent a2ccd85 commit 11cf040

File tree

2 files changed

+1
-4
lines changed

2 files changed

+1
-4
lines changed

tests/integ/test_feature_store.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -312,4 +312,4 @@ def cleanup_feature_group(feature_group: FeatureGroup):
312312
try:
313313
feature_group.delete()
314314
except Exception:
315-
raise RuntimeError(f"Failed to delete feature group with name {feature_group.name}")
315+
raise RuntimeError(f"Failed to delete feature group with name {feature_group.name}")

tests/integ/test_processing.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -288,9 +288,6 @@ def test_sklearn_with_custom_default_bucket(
288288
assert job_description["ProcessingInputs"][0]["InputName"] == "dummy_input"
289289
assert custom_bucket_name in job_description["ProcessingInputs"][0]["S3Input"]["S3Uri"]
290290

291-
assert job_description["ProcessingInputs"][2]["InputName"] == "entrypoint"
292-
assert custom_bucket_name in job_description["ProcessingInputs"][2]["S3Input"]["S3Uri"]
293-
294291
assert job_description["ProcessingJobName"].startswith("test-sklearn-with-customizations")
295292

296293
assert job_description["ProcessingJobStatus"] == "Completed"

0 commit comments

Comments
 (0)