From d9a0ccc9588e771f1f0a7799ec688fe5e31458ec Mon Sep 17 00:00:00 2001 From: Gene Date: Mon, 4 Dec 2023 13:46:47 +0800 Subject: [PATCH] [FIX] fix details --- learnware/specification/regular/image/rkme.py | 2 +- learnware/specification/regular/table/rkme.py | 2 +- learnware/specification/system/hetero_table.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) 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__})!" )