diff --git a/requirements.txt b/requirements.txt
index 820c7a2..8cc7819 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -21,6 +21,7 @@ seaborn
 sparse
 scipy
 sqlalchemy
+pymysql
 xxhash
 wandb
 torch
diff --git a/setup.py b/setup.py
index 2f06cd2..859091c 100644
--- a/setup.py
+++ b/setup.py
@@ -81,6 +81,7 @@ def read_requirements():
     'eva_decord',
     'wandb',
     'sqlalchemy',
+    'pymysql',
 ]}
 
 deps_advanced = {dep: deps_all_dict[dep] for dep in [