File tree Expand file tree Collapse file tree 1 file changed +0
-16
lines changed Expand file tree Collapse file tree 1 file changed +0
-16
lines changed Original file line number Diff line number Diff line change @@ -602,22 +602,6 @@ def inf_instance_family(inf_instance_type):
602
602
return "_" .join (inf_instance_type .split ("." )[0 :2 ])
603
603
604
604
605
- @pytest .fixture (scope = "session" )
606
- def imagenet_train_set (request , sagemaker_session , tmpdir_factory ):
607
- """
608
- Copies the Imagenet dataset from the bucket it's hosted in to the local bucket in the test region.
609
- Due to licensing issues, access to this dataset is controlled through an allowlist
610
- """
611
- local_path = tmpdir_factory .mktemp ("imagenet_tfrecords_train_set" )
612
- sagemaker_session .download_data (
613
- path = local_path , bucket = "collection-of-ml-datasets" , key_prefix = "Imagenet/TFRecords/train"
614
- )
615
- train_input = sagemaker_session .upload_data (
616
- path = local_path , key_prefix = "integ-test-data/imagenet/TFRecords/train"
617
- )
618
- return train_input
619
-
620
-
621
605
def pytest_generate_tests (metafunc ):
622
606
if "instance_type" in metafunc .fixturenames :
623
607
boto_config = metafunc .config .getoption ("--boto-config" )
You can’t perform that action at this time.
0 commit comments