Fix secrets

model_utils.py

@@ -30,11 +30,11 @@ def load_gnn() -> Tuple[torch.nn.Module, torch.nn.Module, torch.nn.Module, torch
                 if not filepath_or_buffer.startswith('s3://'):
                     filepath_or_buffer = 's3://' + filepath_or_buffer
                 kwargs['storage_options'] = {
-                    'key': st.secrets.s3.aws_access_key_id,
-                    'secret': st.secrets.s3.aws_secret_access_key,
+                    'key': st.secrets.aws_access_key_id,
+                    'secret': st.secrets.aws_secret_access_key,
                     'client_kwargs': {
-                        'endpoint_url': st.secrets.s3.endpoint_url,
-                        'region_name': st.secrets.s3.region_name
+                        'endpoint_url': st.secrets.endpoint_url,
+                        'region_name': st.secrets.region_name
                     }
                 }
         return original_read_csv(filepath_or_buffer, *args, **kwargs)
@@ -43,7 +43,7 @@ def load_gnn() -> Tuple[torch.nn.Module, torch.nn.Module, torch.nn.Module, torch
     
     try:
         dataset = MIMIC3Dataset(
-            root=st.secrets.s3.s3_uri,
+            root=st.secrets.s3_uri,
             tables=["DIAGNOSES_ICD","PROCEDURES_ICD","PRESCRIPTIONS","NOTEEVENTS_ICD"],
             code_mapping={"NDC": ("ATC", {"target_kwargs": {"level": 4}})},
         )