Browse Source

[FIX] fix details

tags/v0.3.2
Gene 2 years ago
parent
commit
d9a0ccc958
3 changed files with 3 additions and 3 deletions
  1. +1
    -1
      learnware/specification/regular/image/rkme.py
  2. +1
    -1
      learnware/specification/regular/table/rkme.py
  3. +1
    -1
      learnware/specification/system/hetero_table.py

+ 1
- 1
learnware/specification/regular/image/rkme.py View File

@@ -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__})!"
)


+ 1
- 1
learnware/specification/regular/table/rkme.py View File

@@ -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__})!"
)


+ 1
- 1
learnware/specification/system/hetero_table.py View File

@@ -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__})!"
)


Loading…
Cancel
Save