Skip to content

Commit

Permalink
returned back getters for S3
Browse files Browse the repository at this point in the history
  • Loading branch information
blublinsky committed Jun 6, 2024
1 parent 3ca150e commit 87aa4e1
Showing 1 changed file with 14 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,7 @@ def __init__(
if (s3_credentials is None or s3_credentials.get("access_key", None) is None
or s3_credentials.get("secret_key", None) is None):
raise "S3 credentials is not defined"
self.s3_credentials = s3_credentials
if s3_config is None:
self.input_folder = None
self.output_folder = None
Expand All @@ -69,6 +70,19 @@ def __init__(
region=s3_credentials.get("region", None),
)


def get_access_key(self):
return self.s3_credentials.get("access_key", None)

def get_secret_key(self):
return self.s3_credentials.get("secret_key", None)

def get_endpoint(self):
return self.s3_credentials.get("url", None)

def get_region(self):
return self.s3_credentials.get("region", None)

def get_num_samples(self) -> int:
"""
Get number of samples for input
Expand Down

0 comments on commit 87aa4e1

Please sign in to comment.