File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed
art/estimators/classification Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change 1919This module implements mixin abstract base classes defining properties for all classifiers in ART.
2020"""
2121from abc import ABC , ABCMeta , abstractmethod
22- from typing import List , Optional , Union
22+ from typing import List , Optional , TYPE_CHECKING , Union
2323
2424import numpy as np
2525
2929 LossGradientsMixin ,
3030 DecisionTreeMixin ,
3131)
32- from art .utils import CLASSIFIER_TYPE
32+
33+ if TYPE_CHECKING :
34+ from art .utils import CLASSIFIER_TYPE
3335
3436
3537class InputFilter (ABCMeta ):
@@ -118,7 +120,7 @@ def nb_classes(self, nb_classes: int):
118120
119121 self ._nb_classes = nb_classes
120122
121- def clone_for_refitting (self ) -> CLASSIFIER_TYPE :
123+ def clone_for_refitting (self ) -> " CLASSIFIER_TYPE" :
122124 raise NotImplementedError
123125
124126
You can’t perform that action at this time.
0 commit comments