Skip to content

Commit

Permalink
fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
dtrizna committed Jan 19, 2024
1 parent 7c9853a commit 3a51a08
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion nebula/lit_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -431,7 +431,7 @@ def predict_lit_model(
dump_logits: Union[bool, str] = False
) -> np.ndarray:
assert self.lit_model is not None,\
"[-] lightning_model isn't instantiated: either .train_lit_model() or .load_list_model()"
"[-] lightning_model isn't instantiated: either .train_lit_model() or .load_lit_model()"
"""Get scores out of a loader."""
y_pred_logits = self.trainer.predict(model=self.lit_model, dataloaders=loader)
y_pred = torch.sigmoid(torch.cat(y_pred_logits, dim=0)).numpy()
Expand Down

0 comments on commit 3a51a08

Please sign in to comment.