From 13dfe4e74446f3d1d7782db353f8aff0fd9de97d Mon Sep 17 00:00:00 2001 From: Masaki Kozuki Date: Thu, 7 Nov 2024 21:14:08 +0900 Subject: [PATCH] import cleanup Signed-off-by: Masaki Kozuki --- thunder/tests/test_tensor_subclass.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/thunder/tests/test_tensor_subclass.py b/thunder/tests/test_tensor_subclass.py index ff9643072e..17f3cf0e07 100644 --- a/thunder/tests/test_tensor_subclass.py +++ b/thunder/tests/test_tensor_subclass.py @@ -1,13 +1,12 @@ from __future__ import annotations from typing import TYPE_CHECKING -import pytest import torch from torch.utils import _pytree as pytree import thunder from thunder.core.proxies import SubclassTensorProxy -from thunder.tests.framework import instantiate, nvFuserExecutor +from thunder.tests.framework import instantiate from thunder.tests.make_tensor import make_tensor if TYPE_CHECKING: