diff --git a/learnware/specification/regular/image/rkme.py b/learnware/specification/regular/image/rkme.py index 821924b..448ae62 100644 --- a/learnware/specification/regular/image/rkme.py +++ b/learnware/specification/regular/image/rkme.py @@ -391,7 +391,7 @@ class RKMEImageSpecification(RegularStatSpecification): self.beta = self.beta.to(self._device) self.z = self.z.to(self._device) - if self.type == self.__class__.__name__: + if self.type != self.__class__.__name__: logger.error( f"The type of loaded RKME ({self.type}) is different from the expected type ({self.__class__.__name__})!" ) diff --git a/learnware/specification/regular/table/rkme.py b/learnware/specification/regular/table/rkme.py index af8ff20..78e4996 100644 --- a/learnware/specification/regular/table/rkme.py +++ b/learnware/specification/regular/table/rkme.py @@ -458,7 +458,7 @@ class RKMETableSpecification(RegularStatSpecification): if d in rkme_load.keys(): setattr(self, d, rkme_load[d]) - if self.type == self.__class__.__name__: + if self.type != self.__class__.__name__: logger.error( f"The type of loaded RKME ({self.type}) is different from the expected type ({self.__class__.__name__})!" ) diff --git a/learnware/specification/system/hetero_table.py b/learnware/specification/system/hetero_table.py index ad9d7a8..5bec254 100644 --- a/learnware/specification/system/hetero_table.py +++ b/learnware/specification/system/hetero_table.py @@ -137,7 +137,7 @@ class HeteroMapTableSpecification(SystemStatSpecification): if d in embedding_load.keys(): setattr(self, d, embedding_load[d]) - if self.type == self.__class__.__name__: + if self.type != self.__class__.__name__: logger.error( f"The type of loaded RKME ({self.type}) is different from the expected type ({self.__class__.__name__})!" )