From 20a03f9fe7cf5a232aeea615eb05ac1fdff42a28 Mon Sep 17 00:00:00 2001
From: John Kirkham <jakirkham@gmail.com>
Date: Mon, 16 Mar 2020 16:25:55 -0700
Subject: [PATCH] Import `tlz` for optional `cytoolz` support

---
 CHANGELOG.md                           | 1 +
 python/dask_cudf/dask_cudf/accessor.py | 2 +-
 python/dask_cudf/dask_cudf/core.py     | 2 +-
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index fc38ac15cbf..178df1cb738 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -168,6 +168,7 @@
 - PR #4461 Port nvstrings Miscellaneous functions to cuDF Python/Cython
 - PR #4503 Port binaryop.pyx to libcudf++ API
 - PR #4499 Adding changes to handle include `keep_index` and `RangeIndex`
+- PR #4533 Import `tlz` for optional `cytoolz` support
 - PR #4493 Skip legacy testing in CI
 
 ## Bug Fixes
diff --git a/python/dask_cudf/dask_cudf/accessor.py b/python/dask_cudf/dask_cudf/accessor.py
index eb6e50ea1ad..fceb2c74470 100644
--- a/python/dask_cudf/dask_cudf/accessor.py
+++ b/python/dask_cudf/dask_cudf/accessor.py
@@ -12,7 +12,7 @@
 
 """
 
-from toolz import partial
+from tlz import partial
 
 import cudf
 from cudf.core.column.categorical import (
diff --git a/python/dask_cudf/dask_cudf/core.py b/python/dask_cudf/dask_cudf/core.py
index b245970173d..549e319fc90 100644
--- a/python/dask_cudf/dask_cudf/core.py
+++ b/python/dask_cudf/dask_cudf/core.py
@@ -5,7 +5,7 @@
 
 import numpy as np
 import pandas as pd
-from toolz import partition_all
+from tlz import partition_all
 
 import dask
 import dask.dataframe as dd