adding validations

deepanker13 · Dec 13, 2023 · 34e593e · 34e593e
1 parent 326763e
commit 34e593e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/sdk/python/kubeflow/storage_init_container/s3.py b/sdk/python/kubeflow/storage_init_container/s3.py
@@ -25,6 +25,12 @@ def is_valid_url(self, url):
 
     def __post_init__(self):
         # Custom checks or validations can be added here
+        if (
+            self.bucket_name is None
+            or self.endpoint_url is None
+            or self.file_key is None
+        ):
+            raise ValueError("bucket_name or endpoint_url or file_key is None")
         self.is_valid_url(self.endpoint_url)