[sckip-ci][tmva] Add tensorflow and torch in requirements.txt

root-project · Apr 20, 2024 · c9d8308 · c9d8308
1 parent f1d9fcb
commit c9d8308
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -5,6 +5,8 @@
 numpy>=1.4.1
 scikit-learn
 xgboost
+tensorflow
+torch
 
 # PyROOT: ROOT.Numba.Declare decorator
 numba>=0.47.0 ; python_version < "3.11" # See https://github.com/numba/numba/issues/8304