Skip to content

Commit 2981fa2

Browse files
Improve type checking enforcement
Co-authored-by: Miles Cranmer <miles.cranmer@gmail.com>
1 parent ab33b03 commit 2981fa2

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

scripts/huggingface/upload_model.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -53,9 +53,8 @@ def main(cfg: DictConfig):
5353
dim_out=n_output_fields,
5454
_convert_="all",
5555
)
56-
assert isinstance(model, torch.nn.Module) and isinstance(
57-
model, PyTorchModelHubMixin
58-
)
56+
assert isinstance(model, torch.nn.Module)
57+
assert isinstance(model, PyTorchModelHubMixin)
5958

6059
logger.info(f"Load checkpoints {cfg.model_ckpt}")
6160
checkpoint = torch.load(cfg.model_ckpt, map_location="cpu")

0 commit comments

Comments
 (0)