diff --git a/setup.py b/setup.py
index 427d0b0025..625498cb77 100644
--- a/setup.py
+++ b/setup.py
@@ -38,7 +38,6 @@
     "lightgbm>=2.2.1",
     "memory_profiler>=0.54.0,<1",
     "nltk>=3.4,<4",
-    "pydocumentdb>=2.3.3<3",  # TODO: replace with azure-cosmos
     "seaborn>=0.8.1,<1",
     "transformers>=2.5.0,<5",
     "bottleneck>=1.2.1,<2",