Skip to content

Commit 96e2bb0

Browse files
committed
[fix] Fix mypy issues raised by github check
1 parent 6b28b42 commit 96e2bb0

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
@@ -782,7 +782,7 @@ def _search(
782782
":{}".format(self.task_type, dataset.task_type))
783783

784784
# Initialise information needed for the experiment
785-
experiment_task_name = 'runSearch'
785+
experiment_task_name: str = 'runSearch'
786786
dataset_requirements = get_dataset_requirements(
787787
info=self._get_required_dataset_properties(dataset))
788788
self._dataset_requirements = dataset_requirements
@@ -916,7 +916,7 @@ def _search(
916916
self._stopwatch.stop_task(ensemble_task_name)
917917

918918
# ==> Run SMAC
919-
smac_task_name = 'runSMAC'
919+
smac_task_name: str = 'runSMAC'
920920
self._stopwatch.start_task(smac_task_name)
921921
elapsed_time = self._stopwatch.wall_elapsed(experiment_task_name)
922922
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)