@@ -118,6 +118,8 @@ class ClearMLLogger(BaseLogger):
118118
119119 """
120120
121+ _task : Any
122+
121123 def __init__ (self , ** kwargs : Any ):
122124 try :
123125 from clearml import Task
@@ -823,6 +825,8 @@ class ClearMLSaver(DiskSaver):
823825
824826 """
825827
828+ _task : Any
829+
826830 def __init__ (
827831 self ,
828832 logger : Optional [ClearMLLogger ] = None ,
@@ -858,7 +862,7 @@ def _setup_check_clearml(self, logger: ClearMLLogger, output_uri: str) -> None:
858862 except ImportError :
859863 try :
860864 # Backwards-compatibility for legacy Trains SDK
861- from trains import Task
865+ from trains import Task # type: ignore[no-redef]
862866 except ImportError :
863867 raise ModuleNotFoundError (
864868 "This contrib module requires clearml to be installed. "
@@ -933,7 +937,7 @@ def __call__(self, checkpoint: Mapping, filename: str, metadata: Optional[Mappin
933937 except ImportError :
934938 try :
935939 # Backwards-compatibility for legacy Trains SDK
936- from trains .binding .frameworks import WeightsFileHandler
940+ from trains .binding .frameworks import WeightsFileHandler # type: ignore[no-redef]
937941 except ImportError :
938942 raise ModuleNotFoundError (
939943 "This contrib module requires clearml to be installed. "
@@ -957,8 +961,8 @@ def __call__(self, checkpoint: Mapping, filename: str, metadata: Optional[Mappin
957961 metadata = metadata ,
958962 )
959963
960- pre_cb_id = WeightsFileHandler .add_pre_callback (cb_context .pre_callback )
961- post_cb_id = WeightsFileHandler .add_post_callback (cb_context .post_callback )
964+ pre_cb_id = WeightsFileHandler .add_pre_callback (cb_context .pre_callback ) # type: ignore[arg-type]
965+ post_cb_id = WeightsFileHandler .add_post_callback (cb_context .post_callback ) # type: ignore[arg-type]
962966
963967 try :
964968 super (ClearMLSaver , self ).__call__ (checkpoint , filename , metadata )
0 commit comments