@@ -94,6 +94,7 @@ def fixture_training_job_with_latest_inference_version(
94
94
pytorch .fit ({"training" : _upload_training_data (pytorch )})
95
95
return pytorch .latest_training_job .name
96
96
97
+
97
98
@pytest .mark .skip (
98
99
reason = "The test is temporarily disabled because it's causing errors with 2.4.0 pytorch version. \
99
100
Please run that manually before the proper fix."
@@ -126,6 +127,7 @@ def test_framework_processing_job_with_deps(
126
127
wait = True ,
127
128
)
128
129
130
+
129
131
@pytest .mark .skip (
130
132
reason = "The test is temporarily disabled because it's causing errors with 2.4.0 pytorch version. \
131
133
Please run that manually before the proper fix."
@@ -149,6 +151,7 @@ def test_fit_deploy(
149
151
150
152
assert output .shape == (batch_size , 10 )
151
153
154
+
152
155
@pytest .mark .skip (
153
156
reason = "The test is temporarily disabled because it's causing errors with 2.4.0 pytorch version. \
154
157
Please run that manually before the proper fix."
@@ -179,6 +182,7 @@ def test_local_fit_deploy(
179
182
finally :
180
183
predictor .delete_endpoint ()
181
184
185
+
182
186
@pytest .mark .skip (
183
187
reason = "The test is temporarily disabled because it's causing errors with 2.4.0 pytorch version. \
184
188
Please run that manually before the proper fix."
@@ -213,6 +217,7 @@ def test_deploy_model(
213
217
214
218
assert output .shape == (batch_size , 10 )
215
219
220
+
216
221
@pytest .mark .skip (
217
222
reason = "The test is temporarily disabled because it's causing errors with 2.4.0 pytorch version. \
218
223
Please run that manually before the proper fix."
@@ -243,6 +248,7 @@ def test_deploy_packed_model_with_entry_point_name(
243
248
244
249
assert output .shape == (batch_size , 10 )
245
250
251
+
246
252
@pytest .mark .skip (
247
253
reason = "The test is temporarily disabled because it's causing errors with 2.4.0 pytorch version. \
248
254
Please run that manually before the proper fix."
0 commit comments