  from re import finditer
  from socket import timeout as socket_timeout, error as socket_error
  from threading import get_ident
  from time import time
  from typing import cast, Dict, List, Optional, Union, Tuple, TYPE_CHECKING
  from urllib import request, error as urlliberror
  
  import numpy as np
  from tqdm import tqdm
  
 
</s> remove from typing import Any, cast, List, Tuple, Union
 </s> add  </s> remove from typing import cast, Dict, List, Optional, Tuple, TYPE_CHECKING
 </s> add  </s> remove from typing import cast, Dict, List, Optional, Tuple, TYPE_CHECKING
 </s> add  </s> remove from typing import cast, Callable, Dict, List, Optional, TYPE_CHECKING
 </s> add  </s> remove from typing import cast, Dict, List, Optional, Tuple, TYPE_CHECKING, Union
 </s> add 
  from re import finditer
  from socket import timeout as socket_timeout, error as socket_error
  from threading import get_ident
  from time import time

  from urllib import request, error as urlliberror
  
  import numpy as np
  from tqdm import tqdm
  
keep keep keep keep replace keep keep keep keep keep
 Rebase code (#1326)

* Remove tensorflow_probability requirement

* setup.py - fix progress bars

* requirements.txt: Remove pre python 3.9 packages

* update apple requirements.txt

* update INSTALL.md

* Remove python<3.9 code

* setup.py - fix Windows Installer

* typing: python3.9 compliant

* Update pytest and readthedocs python versions

* typing fixes

* Python Version updates
  - Reduce max version to 3.10
  - Default to 3.10 in installers
  - Remove incompatible 3.11 tests

* Update dependencies

* Downgrade imageio dep for Windows

* typing: merge optional unions and fixes

* Updates
  - min python version 3.10
  - typing to python 3.10 spec
  - remove pre-tf2.10 code
  - Add conda tests

* train: re-enable optimizer saving

* Update dockerfiles

* Update setup.py
  - Apple Conda deps to setup.py
  - Better Cuda + dependency handling

* bugfix: Patch logging to prevent Autograph errors

* Update dockerfiles

* Setup.py - Setup.py - stdout to utf-8

* Add more OSes to github Actions

* suppress mac-os end to end test 
lib/utils.py