Fix where manual live tailing was specifically set to only one user

kerrlabajo · May 16, 2024 · a8f3df8 · a8f3df8
1 parent 4d1c121
commit a8f3df8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/Functions/TrainingJobHandler.cs b/Functions/TrainingJobHandler.cs
@@ -151,10 +151,12 @@ private async Task CheckTrainingJobStatus(AmazonSageMakerClient amazonSageMakerC
                         {
                             if (trainingDetails.SecondaryStatusTransitions.Any() && trainingDetails.SecondaryStatusTransitions.Last().Status == "Training")
                             {
-                                string logStreamName = await GetLatestLogStream(cloudWatchLogsClient, "/aws/sagemaker/TrainingJobs", trainingJobName);
+                                string logGroupName = "/aws/sagemaker/TrainingJobs";
+
+                                string logStreamName = await GetLatestLogStream(cloudWatchLogsClient, logGroupName, trainingJobName);
                                 if (!string.IsNullOrEmpty(logStreamName))
                                 {
-                                    await StartLiveTail(cloudWatchLogsClient, "arn:aws:logs:ap-southeast-1:905418164808:log-group:/aws/sagemaker/TrainingJobs:", logStreamName);
+                                    await StartLiveTail(cloudWatchLogsClient, $"arn:aws:logs:{UserConnectionInfo.Region}:{UserConnectionInfo.AccountId}:log-group:{logGroupName}:", logStreamName);
                                 }
 
                             }