Skip to content

Commit 1ee9d93

Browse files
committed
[fix] Fix mypy issues raised by github check
1 parent f827467 commit 1ee9d93

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

autoPyTorch/api/base_task.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -777,7 +777,7 @@ def _search(
777777
":{}".format(self.task_type, dataset.task_type))
778778

779779
# Initialise information needed for the experiment
780-
experiment_task_name = 'runSearch'
780+
experiment_task_name: str = 'runSearch'
781781
dataset_requirements = get_dataset_requirements(
782782
info=self._get_required_dataset_properties(dataset))
783783
self._dataset_requirements = dataset_requirements
@@ -912,7 +912,7 @@ def _search(
912912
self._stopwatch.stop_task(ensemble_task_name)
913913

914914
# ==> Run SMAC
915-
smac_task_name = 'runSMAC'
915+
smac_task_name: str = 'runSMAC'
916916
self._stopwatch.start_task(smac_task_name)
917917
elapsed_time = self._stopwatch.wall_elapsed(experiment_task_name)
918918
time_left_for_smac = max(0, total_walltime_limit - elapsed_time)

autoPyTorch/datasets/base_dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,7 @@ def __init__(
108108
val_transforms (Optional[torchvision.transforms.Compose]):
109109
Additional Transforms to be applied to the validation/test data
110110
"""
111-
self.dataset_name = dataset_name
111+
self.dataset_name: str = dataset_name
112112

113113
if self.dataset_name is not None:
114114
self.dataset_name = str(uuid.uuid1(clock_seq=os.getpid()))

0 commit comments

Comments
 (0)