diff --git a/dask/base.py b/dask/base.py index 60b0f711ef5..91a8dbab215 100644 --- a/dask/base.py +++ b/dask/base.py @@ -949,6 +949,7 @@ def tokenize(*args, **kwargs): complex, type(Ellipsis), datetime.date, + datetime.timedelta, ), identity, ) diff --git a/dask/tests/test_base.py b/dask/tests/test_base.py index 709c745e933..69bf154359e 100644 --- a/dask/tests/test_base.py +++ b/dask/tests/test_base.py @@ -426,6 +426,11 @@ def test_tokenize_ordered_dict(): assert tokenize(a) != tokenize(c) +def test_tokenize_timedelta(): + assert tokenize(datetime.timedelta(days=1)) == tokenize(datetime.timedelta(days=1)) + assert tokenize(datetime.timedelta(days=1)) != tokenize(datetime.timedelta(days=2)) + + @pytest.mark.parametrize("enum_type", [Enum, IntEnum, IntFlag, Flag]) def test_tokenize_enum(enum_type): class Color(enum_type):