add missing import

Signed-off-by: Evelina <[email protected]>
NVIDIA · Jul 19, 2023 · 92517ac · 92517ac
1 parent 45ffd19
commit 92517ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/collections/nlp/models/nlp_model.py b/nemo/collections/nlp/models/nlp_model.py
@@ -16,7 +16,7 @@
 import hashlib
 import json
 import os
-from typing import Any, Optional
+from typing import Any, Optional, Mapping
 
 from omegaconf import DictConfig, OmegaConf
 from pytorch_lightning import Trainer