Skip to content

Commit c4e9428

Browse files
author
Roja Reddy Sareddy
committed
Add numpy 2.0 support
1 parent 181bd3e commit c4e9428

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

tests/integ/sagemaker/remote_function/test_decorator.py

Lines changed: 1 addition & 3 deletions
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
@@ -316,8 +315,7 @@ def divide(x, y):
316315
divide(10, 2)
317316

318317

319-
@pytest.mark.skipif(
320-
np.__version__ >= "2.0",
318+
@pytest.mark.skip(
321319
reason="Test only valid for numpy < 2.0 due to serialization compatibility changes",
322320
)
323321
def test_with_incompatible_dependencies(

0 commit comments

Comments
 (0)