From ff50317e2d73118761ac189d331ce97cdd89f9ce Mon Sep 17 00:00:00 2001 From: Masaki Kozuki Date: Sat, 28 Dec 2024 13:22:13 +0900 Subject: [PATCH] import cleanup Signed-off-by: Masaki Kozuki --- thunder/clang/__init__.py | 31 +++++++++++++------------------ 1 file changed, 13 insertions(+), 18 deletions(-) diff --git a/thunder/clang/__init__.py b/thunder/clang/__init__.py index 74b1456149..9add31ae34 100644 --- a/thunder/clang/__init__.py +++ b/thunder/clang/__init__.py @@ -1,36 +1,31 @@ -import math +from __future__ import annotations +from collections import namedtuple +from collections.abc import Callable, Sequence from functools import partial, reduce from numbers import Number -from typing import Union, List, Optional, Any -from collections.abc import Callable -from collections.abc import Sequence -from collections import namedtuple +from types import EllipsisType +from typing import Any, Union +import math import operator -from types import EllipsisType, NoneType -import copy -import time import warnings -from thunder.core.baseutils import run_once -from thunder.core.compile_data import using_symbolic_values from thunder.clang.langctx import register_method +from thunder.core import utils +from thunder.core.baseutils import run_once from thunder.core.langctxs import langctx, Languages - +import thunder.core.devices as devices import thunder.core.dtypes as dtypes -from thunder.core import utils import thunder.core.prims as prims from thunder.core.proxies import ( + AnyProxy, IntegerProxy, - NumberProxy, NumberLike, + NumberProxy, + Proxy, TensorProxy, - pyval, pytype, - proxy, - AnyProxy, - Proxy, + pyval, ) -import thunder.core.devices as devices # This file defines the operations in thunder.jit's "core" language. #