class_docstr: `ModelStep` for SageMaker Pipelines Workflows.
class_docstr_tok: ["`","ModelStep","`","for","SageMaker","Pipelines","Workflows","."]
docstr: Assign and validate retry policies according to each kind of sub steps
  
  Args:
      retry_policies (List[RetryPolicy] or Dict[str, List[RetryPolicy]]): The list of retry
          policies for the `ModelStep`.
docstr_tok: ["Assign","and","validate","retry","policies","according","to","each","kind","of","sub","steps","Args",":","retry_policies","(","List","[","RetryPolicy","]","or","Dict","[","str",",","List","[","RetryPolicy","]","]",")",":","The","list","of","retry","policies","for","the","`","ModelStep","`","."]
code: def _assign_and_validate_retry_policies(self, retry_policies):
      """Assign and validate retry policies according to each kind of sub steps
  
      Args:
          retry_policies (List[RetryPolicy] or Dict[str, List[RetryPolicy]]): The list of retry
              policies for the `ModelStep`.
      """
      if isinstance(retry_policies, dict):
          self._create_model_retry_policies = retry_policies.get(
              _CREATE_MODEL_RETRY_POLICIES, None
          )
          self._register_model_retry_policies = retry_policies.get(
              _REGISTER_MODEL_RETRY_POLICIES, None
          )
          self._repack_model_retry_policies = retry_policies.get(
              _REPACK_MODEL_RETRY_POLICIES, None
          )
      else:
          self._create_model_retry_policies = retry_policies
          self._register_model_retry_policies = retry_policies
          self._repack_model_retry_policies = retry_policies
  
      self._validate_sagemaker_job_step_retry_policy()
  
file_path: sagemaker-python-sdk/src/sagemaker/workflow/model_step.py
imports: from __future__ import absolute_import
  import logging
  from typing import Union, List, Dict, Optional
  from sagemaker import Model, PipelineModel, Session
  from sagemaker.workflow._utils import _RegisterModelStep, _RepackModelStep
  from sagemaker.workflow.pipeline_context import PipelineSession, _ModelStepArguments
  from sagemaker.workflow.retry import RetryPolicy, SageMakerJobStepRetryPolicy
  from sagemaker.workflow.step_collections import StepCollection
  from sagemaker.workflow.steps import Step, CreateModelStep
label_desc: Use image node_id 2 for calling the ModelStep obj's underlying member method code with example usage: obj._assign_and_validate_retry_policies(retry_policies) without return types