Moving to SageMaker defaults

Author: ivan-khvostishkov

tests/test_clean_core.py

@@ -1,4 +1,5 @@
 import logging
+import os
 
 import pytest
 import sagemaker
@@ -11,12 +12,11 @@
 
 
 # noinspection DuplicatedCode
-def test_clean_train_warm_pool(request):
+def test_clean_train_warm_pool():
     logging.info("Starting training")
 
-    estimator = PyTorch(entry_point='train_clean.py',
+    estimator = PyTorch(entry_point=os.path.basename('source_dir/training_clean/train_clean.py'),
                         source_dir='source_dir/training_clean/',
-                        role=request.config.getini('sagemaker_role'),
                         framework_version='1.9.1',
                         py_version='py38',
                         instance_count=1,

tests/test_functions.py

@@ -1,5 +1,8 @@
+import logging
+
 import boto3
 import pytest
+import sagemaker.config
 
 from sagemaker.pytorch import PyTorch
 from sagemaker_ssh_helper.wrapper import SSHEnvironmentWrapper, SSHEstimatorWrapper
@@ -102,3 +105,18 @@ def test_bucket_exists():
     _ = _create_bucket_if_doesnt_exist('eu-west-1', custom_bucket_name)
     bucket = _create_bucket_if_doesnt_exist('eu-west-1', custom_bucket_name)
     bucket.delete()
+
+
+def test_sagemaker_default_config_location():
+    f"""
+    See: https://sagemaker.readthedocs.io/en/stable/overview.html#default-configuration-file-location
+    See: {sagemaker.config.config_schema.SAGEMAKER_PYTHON_SDK_CONFIG_SCHEMA}
+    """
+    import os
+    from platformdirs import site_config_dir, user_config_dir
+
+    # Prints the location of the admin config file
+    logging.info(os.path.join(site_config_dir("sagemaker"), "config.yaml"))
+
+    # Prints the location of the user config file
+    logging.info(os.path.join(user_config_dir("sagemaker"), "config.yaml"))