fix pytest

Signed-off-by: Your Name <[email protected]>
kubeflow · Jan 17, 2025 · edfa973 · edfa973
1 parent ebe081c
commit edfa973
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/cmd/suggestion/optuna/v1beta1/requirements.txt b/cmd/suggestion/optuna/v1beta1/requirements.txt
@@ -2,3 +2,5 @@ grpcio>=1.64.1
 protobuf>=4.21.12,<5
 googleapis-common-protos==1.53.0
 optuna==3.6.0
+torch==2.5.1
+scipy==1.15.1