diff --git a/tests/unit/toolkit/test_pretrained_saes.py b/tests/unit/toolkit/test_pretrained_saes.py index 15e18e02..05a0d8a7 100644 --- a/tests/unit/toolkit/test_pretrained_saes.py +++ b/tests/unit/toolkit/test_pretrained_saes.py @@ -1,7 +1,6 @@ import pathlib import shutil -import pytest import torch from sae_lens.toolkit import pretrained_saes