fixed bug in env-vars where sagaemaker model exists but primarycontai…

…ner does not
BishopFox · Mar 27, 2023 · 8ef71a6 · 8ef71a6
1 parent 3652866
commit 8ef71a6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/aws/env-vars.go b/aws/env-vars.go
@@ -750,7 +750,10 @@ func (m *EnvsModule) getSagemakerEnvironmentVariablesPerRegion(r string, wg *syn
 				m.CommandCounter.Error++
 				break
 			}
-			if err == nil {
+
+			// find the number of environment variables in the model in a pointer safe way
+
+			if DescribeModel.PrimaryContainer != nil {
 
 				if len(DescribeModel.PrimaryContainer.Environment) > 0 {
 					name := fmt.Sprintf("[Model] %s", aws.ToString(DescribeModel.ModelName))