Skip to content

Commit

Permalink
#156 unit test related, fixed bug with labels.
Browse files Browse the repository at this point in the history
  • Loading branch information
nicolay-r committed Jan 21, 2023
1 parent 395671f commit f918de2
Showing 1 changed file with 1 addition and 17 deletions.
18 changes: 1 addition & 17 deletions tests/contrib/networks/labels.py
Original file line number Diff line number Diff line change
@@ -1,25 +1,9 @@
from collections import OrderedDict

from arekit.common.labels.base import Label
from arekit.common.labels.scaler.base import BaseLabelScaler


class Label(object):

def __eq__(self, other):
assert(isinstance(other, Label))
return type(self) == type(other)

def __ne__(self, other):
assert(isinstance(other, Label))
return type(self) != type(other)

def __hash__(self):
return hash(self.to_class_str())

def to_class_str(self):
return self.__class__.__name__


class NoLabel(Label):
pass

Expand Down

0 comments on commit f918de2

Please sign in to comment.