From 01800ec3bb17ca169f5567310fcbd6d0301c80a8 Mon Sep 17 00:00:00 2001
From: rchan <rchan@turing.ac.uk>
Date: Fri, 25 Aug 2023 17:11:34 +0100
Subject: [PATCH] update type hint for seqsignet path

---
 src/nlpsig/data_preparation.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/nlpsig/data_preparation.py b/src/nlpsig/data_preparation.py
index 58adeb2..6404082 100644
--- a/src/nlpsig/data_preparation.py
+++ b/src/nlpsig/data_preparation.py
@@ -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.