Skip to content

Commit 06845e5

Browse files
committed
Add report missing imports to allow CI build to run on macOS
1 parent dd4d63c commit 06845e5

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

machine/translation/huggingface/hugging_face_nmt_model_trainer.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -7,14 +7,14 @@
77
from typing import Any, Callable, Dict, List, Optional, Tuple, Union, cast
88

99
import torch # pyright: ignore[reportMissingImports]
10-
from accelerate import Accelerator
11-
from accelerate.utils.memory import should_reduce_batch_size
10+
from accelerate import Accelerator # pyright: ignore[reportMissingImports]
11+
from accelerate.utils.memory import should_reduce_batch_size # pyright: ignore[reportMissingImports]
1212
from datasets.arrow_dataset import Dataset
1313
from sacremoses import MosesPunctNormalizer
1414
from torch import Tensor # pyright: ignore[reportMissingImports]
15-
from torch.nn import Module
16-
from torch.optim.lr_scheduler import LambdaLR
17-
from torch.optim.optimizer import Optimizer
15+
from torch.nn import Module # pyright: ignore[reportMissingImports]
16+
from torch.optim.lr_scheduler import LambdaLR # pyright: ignore[reportMissingImports]
17+
from torch.optim.optimizer import Optimizer # pyright: ignore[reportMissingImports]
1818
from torch.utils.checkpoint import checkpoint # pyright: ignore[reportMissingImports] # noqa: F401
1919
from transformers import (
2020
AutoConfig,

0 commit comments

Comments
 (0)