remove date case

better · Jun 12, 2018 · d7067a3 · d7067a3
1 parent 7bd99dc
commit d7067a3
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/convoys/utils.py b/convoys/utils.py
@@ -41,11 +41,10 @@ def get_groups(data, group_min_size, max_groups):
 
 def _sub(a, b):
     # Computes a - b for a bunch of different cases
-    if isinstance(a, datetime.date):
-        return a - b
-    elif isinstance(a, datetime.datetime) and a.tzinfo is not None:
+    if isinstance(a, datetime.datetime) and a.tzinfo is not None:
         return a.astimezone(b.tzinfo) - b
     else:
+        # Either naive timestamps or numerical type
         return a - b