Skip to content

Commit 55b3dd6

Browse files
authored
fix: AdaptivePredictor.load() (#55)
1 parent ab63cd0 commit 55b3dd6

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

autointent/modules/prediction/_adaptive.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -144,9 +144,13 @@ def load(self, path: str) -> None:
144144
with (dump_dir / self.metadata_dict_name).open() as file:
145145
metadata: AdaptivePredictorDumpMetadata = json.load(file)
146146

147+
if metadata["tags"] is not None and isinstance(metadata["tags"], list):
148+
self.tags = [Tag(**tag) for tag in metadata["tags"]] # type: ignore[arg-type]
149+
else:
150+
self.tags = None
151+
147152
self._r = metadata["r"]
148153
self.n_classes = metadata["n_classes"]
149-
self.tags = [Tag(**tag) for tag in metadata["tags"] if metadata["tags"] and isinstance(metadata["tags"], list)] # type: ignore[arg-type, union-attr]
150154
self.metadata = metadata
151155

152156

0 commit comments

Comments
 (0)