File size: 631 Bytes
de4ade4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
# Copyright 2022 MosaicML LLM Foundry authors
# SPDX-License-Identifier: Apache-2.0

from composer.core import Callback

__all__ = ['ModelGauntlet']


class ModelGauntlet(Callback):
    """The ModelGauntlet callback has been renamed to EvalGauntlet.

    We've created this dummy class, in order to alert anyone who may have been
    importing ModelGauntlet.
    """

    def __init__(
        self,
        *args,  # pyright: ignore [reportMissingParameterType]
        **kwargs):  # pyright: ignore [reportMissingParameterType]
        raise ImportError(
            'ModelGauntlet class is deprecated, please use EvalGauntlet')