Use Python 3.10+ typing in setup and list_versions (#5847)

Summary:
X-link: https://github.com/facebookresearch/FBGEMM/pull/2770

Pull Request resolved: #5847

Reviewed By: cthi

Differential Revision: D107830046

Pulled By: q10

fbshipit-source-id: 5f89c81beeeb805bbe4ac6f3effa233fb12b9b88

Author: cyyever

fbgemm_gpu/list_versions/cli_run.py

@@ -10,7 +10,6 @@
 import logging
 import subprocess
 from datetime import datetime
-from typing import Union
 
 import click
 import pandas as pd
@@ -34,7 +33,7 @@ def __init__(
         self._timestamp = timestamp
         self._visible = visible
 
-    def to_dict(self) -> dict[str, Union[int, str]]:
+    def to_dict(self) -> dict[str, int | str]:
         return {
             "cli": self._cli,
             "stdout": self._stdout,
@@ -52,7 +51,7 @@ def __init__(self) -> None:
         self._cli_outputs: list[CLIOutput] = [
             CLIOutput(
                 cli="python –c “import torch; print(torch.__version__)”",
-                stdout="{}".format(torch.__version__),
+                stdout=f"{torch.__version__}",
                 stderr="",
                 returncode=0,
                 timestamp=datetime.now().isoformat(),
@@ -62,7 +61,7 @@ def __init__(self) -> None:
 
     def run(
         self,
-        cli: Union[str, list[str]],
+        cli: str | list[str],
         visible: bool = True,
         input: str = "",
         capture_output: bool = True,

fbgemm_gpu/setup.py

@@ -14,7 +14,6 @@
 import textwrap
 from dataclasses import dataclass
 from datetime import date
-from typing import Optional
 
 import setuptools
 import setuptools_git_versioning as gitversion
@@ -114,7 +113,7 @@ def is_fbpkg_build(self) -> bool:
             ]
         )
 
-    def nova_flag(self) -> Optional[int]:
+    def nova_flag(self) -> int | None:
         if "BUILD_FROM_NOVA" in os.environ:
             if str(os.getenv("BUILD_FROM_NOVA")) == "0":
                 return 0
@@ -132,7 +131,7 @@ def nova_non_prebuild_step(self) -> bool:
         # flag.  As such, we skip building in the clean and build wheel steps.
         return self.nova_flag() == 1
 
-    def target(self) -> Optional[str]:
+    def target(self) -> str | None:
         if self.args.build_target == "none":
             return None
         return self.args.build_target
@@ -425,7 +424,7 @@ class CudaUtils:
     """CUDA Utilities"""
 
     @classmethod
-    def nvcc_ok(cls, cuda_home: Optional[str], major: int, minor: int) -> bool:
+    def nvcc_ok(cls, cuda_home: str | None, major: int, minor: int) -> bool:
         if not cuda_home:
             return False
 
@@ -448,7 +447,7 @@ def nvcc_ok(cls, cuda_home: Optional[str], major: int, minor: int) -> bool:
         return result
 
     @classmethod
-    def find_cuda(cls, major: int, minor: int) -> Optional[str]:
+    def find_cuda(cls, major: int, minor: int) -> str | None:
         cuda_home = os.environ.get("CUDA_BIN_PATH")
         if cls.nvcc_ok(cuda_home, major, minor):
             return cuda_home