in sagemaker-pyspark-sdk/src/sagemaker_pyspark/SageMakerModel.py [0:0]
def __init__(self,
endpointInstanceType,
endpointInitialInstanceCount,
requestRowSerializer,
responseRowDeserializer,
existingEndpointName=None,
modelImage=None,
modelPath=None,
modelEnvironmentVariables=None,
modelExecutionRoleARN=None,
endpointCreationPolicy=EndpointCreationPolicy.CREATE_ON_CONSTRUCT,
sagemakerClient=SageMakerClients.create_sagemaker_client(),
prependResultRows=True,
namePolicy=RandomNamePolicy(),
uid=None,
javaObject=None):
super(SageMakerModel, self).__init__()
if modelEnvironmentVariables is None:
modelEnvironmentVariables = {}
if javaObject:
self._java_obj = javaObject
else:
if uid is None:
uid = Identifiable._randomUID()
self._java_obj = self._new_java_obj(
SageMakerModel._wrapped_class,
Option(endpointInstanceType),
Option(endpointInitialInstanceCount),
requestRowSerializer,
responseRowDeserializer,
Option(existingEndpointName),
Option(modelImage),
Option(modelPath),
modelEnvironmentVariables,
Option(modelExecutionRoleARN),
endpointCreationPolicy,
sagemakerClient,
prependResultRows,
namePolicy,
uid
)
self._resetUid(self._call_java("uid"))