Skip to content

Commit 9a527b7

Browse files
committed
Removing unused fixtures
1 parent c6ebb5d commit 9a527b7

File tree

1 file changed

+0
-16
lines changed

1 file changed

+0
-16
lines changed

tests/conftest.py

Lines changed: 0 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -602,22 +602,6 @@ def inf_instance_family(inf_instance_type):
602602
return "_".join(inf_instance_type.split(".")[0:2])
603603

604604

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-
621605
def pytest_generate_tests(metafunc):
622606
if "instance_type" in metafunc.fixturenames:
623607
boto_config = metafunc.config.getoption("--boto-config")

0 commit comments

Comments
 (0)