def set_platform_param()

in model_optimizer_pkg/model_optimizer_pkg/model_optimizer_node.py [0:0]

• 7 lines of code
• 3 McCabe index (conditional complexity)

    def set_platform_param(self, platform_param, aux_inputs):
        """Helper method that creates a dictionary with the platform specific
           Intel model optimizer cli commands.

        Args:
            platform_param (dict): Dictionary of available platform cli commands.
            aux_inputs (dict): Dictionary of auxiliary options for the model optimizer.

        Returns:
            dict: Dictionary with platform specific params set if present in aux_inputs.
        """
        self.get_logger().info(f"aux_inputs: {aux_inputs} ")
        set_paltform_params = {}
        for flag in platform_param:
            if flag in aux_inputs:
                set_paltform_params[flag] = aux_inputs[flag]
        return set_paltform_params