Adding sagemaker resources (#204)

mingrammer · Jun 17, 2020 · c0bb034 · c0bb034
1 parent a0891aa
commit c0bb034
Show file tree

Hide file tree

Showing 5 changed files with 15 additions and 0 deletions.
diff --git a/diagrams/aws/ml.py b/diagrams/aws/ml.py
@@ -64,6 +64,18 @@ class SagemakerGroundTruth(_ML):
     _icon = "sagemaker-ground-truth.png"
 
 
+class SagemakerModel(_ML):
+    _icon = "sagemaker-model.png"
+
+
+class SagemakerNotebook(_ML):
+    _icon = "sagemaker-notebook.png"
+
+
+class SagemakerTrainingJob(_ML):
+    _icon = "sagemaker-training-job.png"
+
+
 class Sagemaker(_ML):
     _icon = "sagemaker.png"
 

diff --git a/docs/nodes/aws.md b/docs/nodes/aws.md
@@ -256,6 +256,9 @@ Node classes list of aws provider.
 - **diagrams.aws.ml.Polly**
 - **diagrams.aws.ml.Rekognition**
 - **diagrams.aws.ml.SagemakerGroundTruth**
+- **diagrams.aws.ml.SagemakerModel**
+- **diagrams.aws.ml.SagemakerNotebook**
+- **diagrams.aws.ml.SagemakerTrainingJob**
 - **diagrams.aws.ml.Sagemaker**
 - **diagrams.aws.ml.TensorflowOnAWS**
 - **diagrams.aws.ml.Textract**

diff --git a/resources/aws/ml/sagemaker-model.png b/resources/aws/ml/sagemaker-model.png
diff --git a/resources/aws/ml/sagemaker-notebook.png b/resources/aws/ml/sagemaker-notebook.png
diff --git a/resources/aws/ml/sagemaker-training-job.png b/resources/aws/ml/sagemaker-training-job.png