Skip to content

Commit

Permalink
update type hint for seqsignet path
Browse files Browse the repository at this point in the history
  • Loading branch information
rchan26 committed Aug 25, 2023
1 parent 35a247b commit 01800ec
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/nlpsig/data_preparation.py
Original file line number Diff line number Diff line change
Expand Up @@ -1211,7 +1211,7 @@ def get_torch_path_for_SeqSigNet(
include_features_in_input: bool,
include_embedding_in_input: bool,
reduced_embeddings: bool = False,
) -> dict[str, torch.tensor | int]:
) -> dict[str, dict[str, torch.tensor] | int | None]:
"""
Returns a `torch.tensor` object that can be passed into `nlpsig_networks.SeqSigNet` model.
Expand Down

0 comments on commit 01800ec

Please sign in to comment.