import cleanup

Signed-off-by: Masaki Kozuki <[email protected]>
Lightning-AI · Dec 7, 2024 · 0f793aa · 0f793aa
1 parent ba97cef
commit 0f793aa
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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: