7919331b412d25e7a02a9447377dd4ee0c847c3a,src/sagemaker/pytorch/model.py,PyTorchModel,__init__,#PyTorchModel#Any#Any#Any#Any#Any#Any#Any#Any#,64

Before Change

                python_deprecation_warning(self.__framework_name__, defaults.LATEST_PY2_VERSION)
            )

        if framework_version is None:
            logger.warning(
                empty_framework_version_warning(defaults.PYTORCH_VERSION, defaults.LATEST_VERSION)
            )

        self.py_version = py_version
        self.framework_version = framework_version or defaults.PYTORCH_VERSION
        self.model_server_workers = model_server_workers

    def prepare_container_def(self, instance_type, accelerator_type=None):
        Return a container definition with framework configuration set in

After Change

            :class:`~sagemaker.model.FrameworkModel` and
            :class:`~sagemaker.model.Model`.
        
        validate_version_or_image_args(framework_version, py_version, image)
        if py_version and py_version == "py2":
            logger.warning(
                python_deprecation_warning(self.__framework_name__, defaults.LATEST_PY2_VERSION)
            )
        self.framework_version = framework_version
        self.py_version = py_version

        super(PyTorchModel, self).__init__(
            model_data, image, role, entry_point, predictor_cls=predictor_cls, **kwargs

Instances