Typing fixes

Author: vivekmig

tests/attr/models/test_pytext.py

@@ -8,7 +8,6 @@
 from typing import Dict, List, NoReturn, Optional
 
 import torch
-from pytext.data.data_handler import CommonMetadata
 
 HAS_PYTEXT = True
 try:
@@ -29,6 +28,7 @@
     from pytext.models.doc_model import DocModel_Deprecated  # @manual=//pytext:main_lib
     from pytext.models.embeddings.word_embedding import WordEmbedding
     from pytext.models.representations.bilstm_doc_attention import BiLSTMDocAttention
+    from pytext.data.data_handler import CommonMetadata
 except ImportError:
     HAS_PYTEXT = False
 

tests/concept/test_tcav.py

@@ -14,7 +14,6 @@
     Iterator,
     List,
     Set,
-    SupportsIndex,
     Tuple,
     Union,
 )
@@ -174,7 +173,7 @@ def __init__(
         self,
         get_tensor_from_filename_func: Callable,
         path: str,
-        num_samples: SupportsIndex = 100,
+        num_samples: int = 100,
     ) -> None:
         r"""
         Args: