From 5f0fd2f9abc07c7cd57dc1150ac55283de4717f7 Mon Sep 17 00:00:00 2001
From: Georgios Efstathiadis <gioefst2000@gmail.com>
Date: Mon, 16 Oct 2023 15:51:47 -0400
Subject: [PATCH] typing fix for return

---
 forest/jasmine/traj2stats.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/forest/jasmine/traj2stats.py b/forest/jasmine/traj2stats.py
index bb1699c8..c15a8704 100644
--- a/forest/jasmine/traj2stats.py
+++ b/forest/jasmine/traj2stats.py
@@ -260,7 +260,7 @@ def get_nearby_locations(
 def avg_mobility_trace_difference(
     time_range: Tuple[int, int], mobility_trace1: np.ndarray,
     mobility_trace2: np.ndarray
-) -> float:
+) -> Optional[float]:
     """This function calculates the average mobility trace difference
 
     Args: