doc: Adding doc strings to tests

Author: Lokiiiiii

tests/conftest.py

@@ -471,7 +471,8 @@ def inf_instance_family(inf_instance_type):
 @pytest.fixture(scope="session")
 def imagenet_train_subset(request, sagemaker_session, tmpdir_factory):
     """
-    Copies the Imagenet dataset from the bucket it's hosted in to the local bucket in the test region
+    Copies the Imagenet dataset from the bucket it's hosted in to the local bucket in the test region.
+    Due to licensing issues, access to this dataset is controlled through an allowlist
     """
     local_path = tmpdir_factory.mktemp("imagenet_tfrecords_train_subset")
     sagemaker_session.download_data(

tests/integ/test_training_compiler.py

@@ -34,7 +34,8 @@ def gpu_instance_type(request):
 @pytest.fixture(scope="module")
 def imagenet_val_set(request, sagemaker_session, tmpdir_factory):
     """
-    Copies the dataset from the bucket it's hosted in to the local bucket in the test region
+    Copies the Imagenet dataset from the bucket it's hosted in to the local bucket in the test region.
+    Due to licensing issues, access to this dataset is controlled through an allowlist
     """
     local_path = tmpdir_factory.mktemp("trcomp_imagenet_val_set")
     sagemaker_session.download_data(