diff --git a/docker/tune_test/requirements.txt b/docker/tune_test/requirements.txt
index a2603736a..413ac27e8 100644
--- a/docker/tune_test/requirements.txt
+++ b/docker/tune_test/requirements.txt
@@ -31,3 +31,4 @@ torchvision
 xgboost
 zoopt>=0.4.0
 timm
+dataclasses