Skip to content

Commit abacd52

Browse files
committed
ignore vulnerable
1 parent 02ef6b8 commit abacd52

File tree

2 files changed

+8
-5
lines changed

2 files changed

+8
-5
lines changed

data/ignore_ids_safety_scan.json

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1386,7 +1386,8 @@
13861386
"65095": "Covered elsewhere",
13871387
"61503": "Covered elsewhere",
13881388
"65193": "Covered elsewhere",
1389-
"62019": "Covered elsehwere"
1389+
"62019": "Covered elsehwere",
1390+
"67599": "only affects the use of --extra-index-url which is not the case"
13901391
}
13911392
},
13921393
"inference": {

test/sagemaker_tests/huggingface_pytorch/training/integration/sagemaker/test_smmp.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525

2626
# hyperparameters, which are passed into the training job
2727
hyperparameters = {
28-
"model_name_or_path": "hf-internal-testing/tiny-random-RobertaModel",
28+
"model_name_or_path": "hf-internal-testing/tiny-random-DistilBertModel",
2929
"task_name": "mnli",
3030
"per_device_train_batch_size": 2, # batch size must be divisible by the number of microbatches
3131
"per_device_eval_batch_size": 2,
@@ -68,7 +68,7 @@ def get_transformers_version_from_image_uri(ecr_image):
6868

6969
@pytest.mark.processor("gpu")
7070
@pytest.mark.integration("smmp")
71-
@pytest.mark.model("hf_qa_smmp")
71+
@pytest.mark.model("hf_distilbert_smmp")
7272
@pytest.mark.skip_cpu
7373
@pytest.mark.skip_py2_containers
7474
@pytest.mark.skip_trcomp_containers
@@ -85,7 +85,7 @@ def test_smmp_gpu(
8585

8686
@pytest.mark.processor("gpu")
8787
@pytest.mark.integration("smmp")
88-
@pytest.mark.model("hf_qa_smmp_multi")
88+
@pytest.mark.model("hf_distilbert_smmp_multi")
8989
@pytest.mark.skip_cpu
9090
@pytest.mark.skip_py2_containers
9191
@pytest.mark.multinode(2)
@@ -126,4 +126,6 @@ def _test_smmp_gpu_function(ecr_image, sagemaker_session, py_version, instances_
126126
hyperparameters=hyperparameters,
127127
sagemaker_session=sagemaker_session,
128128
)
129-
huggingface_estimator.fit(job_name=sagemaker.utils.unique_name_from_base("test-hf-pt-qa-smmp"))
129+
huggingface_estimator.fit(
130+
job_name=sagemaker.utils.unique_name_from_base("test-hf-pt-text-classif-smmp")
131+
)

0 commit comments

Comments
 (0)