Skip to content

Commit

Permalink
Fix test after changing type of classes_
Browse files Browse the repository at this point in the history
  • Loading branch information
BenjaminBossan committed Feb 8, 2024
1 parent d022b23 commit 43f5d3f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion skorch/tests/test_classifier.py
Original file line number Diff line number Diff line change
Expand Up @@ -134,7 +134,7 @@ def test_classes_with_gaps(self, net_cls, module_cls, data):

def test_pass_classes_explicitly_overrides(self, net_cls, module_cls, data):
net = net_cls(module_cls, max_epochs=0, classes=['foo', 'bar']).fit(*data)
assert net.classes_ == ['foo', 'bar']
assert (net.classes_ == np.array(['foo', 'bar'])).all()

def test_classes_are_set_with_tensordataset_explicit_y(
self, net_cls, module_cls, data
Expand Down

0 comments on commit 43f5d3f

Please sign in to comment.