# -*- coding: utf-8 -*-
# (C) Copyright 2020, 2021, 2022, 2023, 2024 IBM. All Rights Reserved.
#
# This code is licensed under the Apache License, Version 2.0. You may
# obtain a copy of this license in the LICENSE.txt file in the root directory
# of this source tree or at http://www.apache.org/licenses/LICENSE-2.0.
#
# Any modifications or derivative works of this code must retain this
# copyright notice, and modified files need to carry a notice indicating
# that they have been altered from the originals.
"""Helper class for adding rpu_config to neural network model"""
# pylint: disable=no-name-in-module,import-error
from aihwkit.cloud.converter.definitions.i_input_file_pb2 import ( # type: ignore[attr-defined]
AnalogProto,
)
# pylint: disable=too-few-public-methods)
[docs]class AnalogInfo:
"""Data class for fields from protobuf AnalogProto message"""
def __init__(self, a_info: AnalogProto): # type: ignore[valid-type]
"""Constructor for this class"""
# all three fields are required
self.output_noise_strength = a_info.output_noise_strength # type: ignore[attr-defined]
self.adc = a_info.adc # type: ignore[attr-defined]
self.dac = a_info.dac # type: ignore[attr-defined]