another isinstance

wireservice · Mar 16, 2020 · 29ccc9d · 29ccc9d
1 parent 173e6d6
commit 29ccc9d
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/agate/computations/change.py b/agate/computations/change.py
@@ -27,11 +27,7 @@ def __init__(self, before_column_name, after_column_name):
     def get_computed_data_type(self, table):
         before_column = table.columns[self._before_column_name]
 
-        if isinstance(before_column.data_type, Date):
-            return TimeDelta()
-        elif isinstance(before_column.data_type, DateTime):
-            return TimeDelta()
-        elif isinstance(before_column.data_type, TimeDelta):
+        if isinstance(before_column.data_type, (Date, DateTime, TimeDelta)):
             return TimeDelta()
         elif isinstance(before_column.data_type, Number):
             return Number()