def fromEndpoint()

in sagemaker-spark-sdk/src/main/scala/com/amazonaws/services/sagemaker/sparksdk/SageMakerModel.scala [209:239]

27 lines of code
1 McCabe index (conditional complexity)


  def fromEndpoint(endpointName: String,
                   requestRowSerializer: RequestRowSerializer,
                   responseRowDeserializer: ResponseRowDeserializer,
                   modelEnvironmentVariables: Map[String, String] = Map[String, String](),
                   sagemakerClient : AmazonSageMaker
                   = AmazonSageMakerClientBuilder.defaultClient,
                   prependResultRows : Boolean = true,
                   namePolicy : NamePolicy = new RandomNamePolicy(),
                   uid: String = Identifiable.randomUID("sagemaker")) : SageMakerModel = {

    val describeEndpointRequest = new DescribeEndpointRequest().withEndpointName(endpointName)
    InternalUtils.applyUserAgent(describeEndpointRequest)
    val result = sagemakerClient.describeEndpoint(describeEndpointRequest)

    val status = result.getEndpointStatus

    require(status == EndpointStatus.InService.toString, "Endpoint status must be In Service, not "
      + status)

    new SageMakerModel(existingEndpointName = Some(endpointName),
      requestRowSerializer = requestRowSerializer,
      responseRowDeserializer = responseRowDeserializer,
      modelEnvironmentVariables = modelEnvironmentVariables,
      endpointCreationPolicy = EndpointCreationPolicy.DO_NOT_CREATE,
      endpointInstanceType = None,
      endpointInitialInstanceCount = None,
      sagemakerClient = sagemakerClient,
      prependResultRows = prependResultRows,
      namePolicy = namePolicy,
      uid = uid)
  }