summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authorMax Ryabinin <mryabinin0@gmail.com>2022-06-30 18:14:20 +0300
committerMax Ryabinin <mryabinin0@gmail.com>2022-06-30 18:14:20 +0300
commit33efe4a09f459832e8beceba70add0695cc485e4 (patch)
tree546ef2dea977f9850b4afeb9bfb18871ef948654 /tests
parent4e60e7dc62c50b6ba9b6becf6e779a1d48906be2 (diff)
Remove unused imports, fix NotImplementedError
Diffstat (limited to 'tests')
-rw-r--r--tests/test_modules.py4
-rw-r--r--tests/test_optim.py1
2 files changed, 0 insertions, 5 deletions
diff --git a/tests/test_modules.py b/tests/test_modules.py
index 6cbee7b..a0379cb 100644
--- a/tests/test_modules.py
+++ b/tests/test_modules.py
@@ -6,10 +6,6 @@ import pytest
import torch
import bitsandbytes as bnb
-from itertools import product
-
-from bitsandbytes import functional as F
-
@pytest.mark.parametrize("embcls", [bnb.nn.Embedding, bnb.nn.StableEmbedding], ids=['Embedding', 'StableEmbedding'])
def test_embeddings(embcls):
diff --git a/tests/test_optim.py b/tests/test_optim.py
index 5464043..c80fe51 100644
--- a/tests/test_optim.py
+++ b/tests/test_optim.py
@@ -7,7 +7,6 @@ import time
import shutil
import uuid
import pytest
-import ctypes
import torch
import bitsandbytes as bnb
import bitsandbytes.functional as F