Update processing_cxrmate2.py

processing_cxrmate2.py

@@ -19,10 +19,8 @@ from transformers.tokenization_utils_base import PreTokenizedInput, TextInput
 
 try:
     from .dataset import CXRMate2Dataset
-    from .utils import compute_time_delta
 except ImportError:
     from dataset import CXRMate2Dataset
-    from utils import compute_time_delta
 
 # Ordered by oblique, lateral, AP, and then PA views so that PA views are closest in position to the generated tokens (and oblique is furtherest).
 VIEW_ORDER = [
@@ -52,6 +50,17 @@ VIEW_ORDER = [
 ]
 
 
+def compute_time_delta(event_time, reference_time, to_tensor=True):
+    time_delta = reference_time - event_time
+    time_delta = time_delta.total_seconds()
+    assert isinstance(time_delta, float), f'time_delta should be float, not {type(time_delta)}.'
+    if time_delta < 0:
+        raise ValueError(f'time_delta should be greater than or equal to zero, not {time_delta}.')
+    if to_tensor: 
+        time_delta = torch.tensor(time_delta)
+    return time_delta
+
+
 class CXRMate2Processor(transformers.ProcessorMixin):
 
     attributes = ['image_processor', 'tokenizer']