  import ctypes
  from ctypes import POINTER, Structure, windll
  from dataclasses import dataclass
  from enum import Enum, IntEnum
  from typing import Any, Callable, cast, List
  
  from comtypes import COMError, IUnknown, GUID, STDMETHOD, HRESULT  # pylint:disable=import-error
  
  from ._base import _GPUStats
  
 
</s> remove from typing import Callable, cast, List, Optional, Tuple, TYPE_CHECKING, Union
 </s> add  </s> remove 
 </s> add from __future__ import annotations </s> remove import sys
 </s> add import typing as T </s> remove from typing import List
 </s> add 
  import ctypes
  from ctypes import POINTER, Structure, windll
  from dataclasses import dataclass
  from enum import Enum, IntEnum

  
  from comtypes import COMError, IUnknown, GUID, STDMETHOD, HRESULT  # pylint:disable=import-error
  
  from ._base import _GPUStats
  
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/gpu_stats/directml.py