def to_dict()

in optimum/graphcore/training_args.py [0:0]

• 10 lines of code
• 8 McCabe index (conditional complexity)

    def to_dict(self):
        """
        Serializes this instance while replacing the `Enum` with their values (for JSON serialization support). It obfuscates
        the token values by removing their value.
        """
        d = asdict(self)
        for k, v in d.items():
            if isinstance(v, Enum):
                d[k] = v.value
            if isinstance(v, list) and len(v) > 0 and isinstance(v[0], Enum):
                d[k] = [x.value for x in v]
            if k.endswith("_token"):
                d[k] = f"<{k.upper()}>"
        return d