From a8571b5c5875a873588418c1f80ceb2ca7f1e349 Mon Sep 17 00:00:00 2001
From: Yuqi Chen <42923882+chenyuqi990215@users.noreply.github.com>
Date: Sun, 26 May 2024 17:53:22 +0800
Subject: [PATCH 1/4] Update tpp.py

---
 physiopro/dataset/tpp.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/physiopro/dataset/tpp.py b/physiopro/dataset/tpp.py
index 54630c9..27540cc 100644
--- a/physiopro/dataset/tpp.py
+++ b/physiopro/dataset/tpp.py
@@ -1,11 +1,13 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 
-from typing import Optional, Any
+from typing import Optional, Any, Union
 import os
 import pickle
 import numpy as np
 import torch
+from utilsd.config import ClassConfig
+from .preprocessor import Preprocessor
 from .base import BaseDataset, DATASETS
 
 
@@ -20,7 +22,7 @@ def __init__(
         name: str = "Neonate",
         fold: str = "folder1",
         dataset: Optional[str] = "train",
-        preprocessor: Optional[Any] = None,
+        preprocessor: Optional[Union[ClassConfig[Preprocessor], Preprocessor]] = None,
         max_len: Optional[int] = 500,
         clip_max: Optional[int] = -1,
     ):

From 539b1e32e013b16504a4e9b123412d160fb1da8f Mon Sep 17 00:00:00 2001
From: Yuqi Chen <42923882+chenyuqi990215@users.noreply.github.com>
Date: Sun, 26 May 2024 18:03:56 +0800
Subject: [PATCH 2/4] Update tpp.py

---
 physiopro/dataset/tpp.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/physiopro/dataset/tpp.py b/physiopro/dataset/tpp.py
index 27540cc..b25fd38 100644
--- a/physiopro/dataset/tpp.py
+++ b/physiopro/dataset/tpp.py
@@ -1,7 +1,7 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 
-from typing import Optional, Any, Union
+from typing import Optional, Union
 import os
 import pickle
 import numpy as np

From 9379211b113d7986b60574f31d4ca41f6cd7ced2 Mon Sep 17 00:00:00 2001
From: Yuqi Chen <42923882+chenyuqi990215@users.noreply.github.com>
Date: Tue, 22 Jul 2025 18:36:24 +0800
Subject: [PATCH 3/4] Update eventprediction.py

---
 physiopro/model/eventprediction.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/physiopro/model/eventprediction.py b/physiopro/model/eventprediction.py
index 6973ece..4895f6b 100644
--- a/physiopro/model/eventprediction.py
+++ b/physiopro/model/eventprediction.py
@@ -409,8 +409,10 @@ def forward(self, inputs):
         if self.temporal_encoding:
             tem_enc = self.temporal_enc(event_time, non_pad_mask)
             enc_output += tem_enc
-
-        enc_out = self.network(enc_output)
+        try:  # for Contiformer, it requires event_time as input
+            enc_out = self.network(enc_output, t=event_time)
+        except:
+            enc_out = self.network(enc_output)
 
         if self.hyper_paras["use_rnn"]:
             enc_out = self.rnn(enc_output, non_pad_mask)

From bcf608339b8a748f8600ab76aa9035c00d1841ff Mon Sep 17 00:00:00 2001
From: Yuqi Chen <42923882+chenyuqi990215@users.noreply.github.com>
Date: Tue, 22 Jul 2025 18:37:06 +0800
Subject: [PATCH 4/4] Update eventprediction.py

---
 physiopro/model/eventprediction.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/physiopro/model/eventprediction.py b/physiopro/model/eventprediction.py
index 4895f6b..aaf0167 100644
--- a/physiopro/model/eventprediction.py
+++ b/physiopro/model/eventprediction.py
@@ -409,6 +409,7 @@ def forward(self, inputs):
         if self.temporal_encoding:
             tem_enc = self.temporal_enc(event_time, non_pad_mask)
             enc_output += tem_enc
+            
         try:  # for Contiformer, it requires event_time as input
             enc_out = self.network(enc_output, t=event_time)
         except: