typing fix for return

onnela-lab · Oct 16, 2023 · 5f0fd2f · 5f0fd2f
1 parent 5ac8e12
commit 5f0fd2f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: