Skip to content

Commit

Permalink
removing transformer from test
Browse files Browse the repository at this point in the history
  • Loading branch information
carsen-stringer committed Apr 7, 2024
1 parent 8c8e9d8 commit 518f92b
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions tests/test_import.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,10 +7,8 @@ def test_cellpose_imports_without_error():
def test_model_zoo_imports_without_error():
from cellpose import models, denoise
for model_name in models.MODEL_NAMES:
if "neurips" not in model_name:
model = models.CellposeModel(model_type=model_name,
backbone="transformer" if "transformer" in model_name else "default")

if "neurips" not in model_name and "transformer" not in model_name:
model = models.CellposeModel(model_type=model_name)

def test_gui_imports_without_error():
from cellpose import gui
Expand Down

0 comments on commit 518f92b

Please sign in to comment.