Merge branch 'master-jumpstart' into feat/client-cache-for-jumpstart-models

Author: evakravi

doc/api/inference/model_monitor.rst

@@ -31,3 +31,13 @@ Model Monitor
     :members:
     :undoc-members:
     :show-inheritance:
+
+.. automodule:: sagemaker.model_metrics
+    :members:
+    :undoc-members:
+    :show-inheritance:
+
+.. automodule:: sagemaker.drift_check_baselines
+    :members:
+    :undoc-members:
+    :show-inheritance:

doc/frameworks/huggingface/sagemaker.huggingface.rst

@@ -4,7 +4,15 @@ Hugging Face
 Hugging Face Estimator
 ----------------------
 
-.. autoclass:: sagemaker.huggingface.estimator.HuggingFace
+.. autoclass:: sagemaker.huggingface.HuggingFace
+    :members:
+    :undoc-members:
+    :show-inheritance:
+
+Hugging Face Training Compiler Configuration
+--------------------------------------------
+
+.. autoclass:: sagemaker.huggingface.TrainingCompilerConfig
     :members:
     :undoc-members:
     :show-inheritance:
@@ -17,8 +25,8 @@ Hugging Face Model
     :undoc-members:
     :show-inheritance:
 
-HuggingFace Predictor
----------------------
+Hugging Face Predictor
+----------------------
 
 .. autoclass:: sagemaker.huggingface.model.HuggingFacePredictor
     :members:

doc/workflows/pipelines/sagemaker.workflow.pipelines.rst

@@ -32,6 +32,11 @@ Conditions
 
 .. autoclass:: sagemaker.workflow.conditions.ConditionOr
 
+CheckJobConfig
+--------------
+
+.. autoclass:: sagemaker.workflow.check_job_config.CheckJobConfig
+
 Entities
 --------
 
@@ -128,3 +133,13 @@ Steps
 .. autoclass:: sagemaker.workflow.steps.CacheConfig
 
 .. autoclass:: sagemaker.workflow.lambda_step.LambdaStep
+
+.. autoclass:: sagemaker.workflow.steps.CompilationStep
+
+.. autoclass:: sagemaker.workflow.quality_check_step.QualityCheckConfig
+
+.. autoclass:: sagemaker.workflow.quality_check_step.QualityCheckStep
+
+.. autoclass:: sagemaker.workflow.clarify_check_step.ClarifyCheckConfig
+
+.. autoclass:: sagemaker.workflow.clarify_check_step.ClarifyCheckStep

setup.py

@@ -34,7 +34,7 @@ def read_version():
 # Declare minimal set for installation
 required_packages = [
     "attrs",
-    "boto3>=1.16.32",
+    "boto3>=1.20.18",
     "google-pasta",
     "numpy>=1.9.0",
     "protobuf>=3.1",
@@ -70,6 +70,7 @@ def read_version():
         "pytest-rerunfailures",
         "pytest-timeout",
         "pytest-xdist",
+        "coverage<6.2",
         "mock",
         "contextlib2",
         "awslogs",

src/sagemaker/__init__.py

@@ -49,7 +49,7 @@
 from sagemaker.local.local_session import LocalSession  # noqa: F401
 
 from sagemaker.model import Model, ModelPackage  # noqa: F401
-from sagemaker.model_metrics import ModelMetrics, MetricsSource  # noqa: F401
+from sagemaker.model_metrics import ModelMetrics, MetricsSource, FileSource  # noqa: F401
 from sagemaker.pipeline import PipelineModel  # noqa: F401
 from sagemaker.predictor import Predictor  # noqa: F401
 from sagemaker.processing import Processor, ScriptProcessor  # noqa: F401