Get all the Environment variables that will be passed to the container

        Certain input fields such as BatchStrategy have different values for the API vs the Environment
        variables, such as SingleRecord vs SINGLE_RECORD. This method also handles this conversion.

        Args:
            **kwargs: existing transform arguments

        Returns:
            dict: All the environment variables that should be set in the container

def _get_container_environment(self, **kwargs):
        """Get all the Environment variables that will be passed to the container

        Certain input fields such as BatchStrategy have different values for the API vs the Environment
        variables, such as SingleRecord vs SINGLE_RECORD. This method also handles this conversion.

        Args:
            **kwargs: existing transform arguments

        Returns:
            dict: All the environment variables that should be set in the container

        """
        environment = {}
        environment.update(self.primary_container['Environment'])
        environment['SAGEMAKER_BATCH'] = 'True'
        if 'MaxPayloadInMB' in kwargs:
            environment['SAGEMAKER_MAX_PAYLOAD_IN_MB'] = str(kwargs['MaxPayloadInMB'])

        if 'BatchStrategy' in kwargs:
            if kwargs['BatchStrategy'] == 'SingleRecord':
                strategy_env_value = 'SINGLE_RECORD'
            elif kwargs['BatchStrategy'] == 'MultiRecord':
                strategy_env_value = 'MULTI_RECORD'
            else:
                raise ValueError('Invalid BatchStrategy, must be \'SingleRecord\' or \'MultiRecord\'')
            environment['SAGEMAKER_BATCH_STRATEGY'] = strategy_env_value

        # we only do 1 max concurrent transform in Local Mode
        if 'MaxConcurrentTransforms' in kwargs and int(kwargs['MaxConcurrentTransforms']) > 1:
            logger.warning('Local Mode only supports 1 ConcurrentTransform. Setting MaxConcurrentTransforms to 1')
        environment['SAGEMAKER_MAX_CONCURRENT_TRANSFORMS'] = '1'

        # if there were environment variables passed to the Transformer we will pass them to the
        # container as well.
        if 'Environment' in kwargs:
            environment.update(kwargs['Environment'])
        return environment