diff --git a/src/sagemaker/local/data.py b/src/sagemaker/local/data.py index 13c966e809..7748eced5b 100644 --- a/src/sagemaker/local/data.py +++ b/src/sagemaker/local/data.py @@ -122,6 +122,8 @@ def __init__(self, root_path): Args: root_path: """ + super(LocalFileDataSource, self).__init__() + self.root_path = os.path.abspath(root_path) if not os.path.exists(self.root_path): raise RuntimeError("Invalid data source: %s does not exist." % self.root_path) @@ -167,6 +169,7 @@ def __init__(self, bucket, prefix, sagemaker_session): desired settings to talk to S3 """ + super(S3DataSource, self).__init__() # Create a temporary dir to store the S3 contents root_dir = sagemaker.utils.get_config_value(