From 737d2eac8677e4c8b4a181764ab1304c311cd493 Mon Sep 17 00:00:00 2001
From: Holger Roth <6304754+holgerroth@users.noreply.github.com>
Date: Thu, 10 Oct 2024 19:29:13 -0400
Subject: [PATCH] FedBPT: Fix fedbpt cma version (#3029) (#3030)

* fix cma version; upgrade nvflare version

* upgrade python to 3.12
---
 research/fed-bpt/README.md        | 2 +-
 research/fed-bpt/requirements.txt | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/research/fed-bpt/README.md b/research/fed-bpt/README.md
index e32a3ea144..67f63449d6 100644
--- a/research/fed-bpt/README.md
+++ b/research/fed-bpt/README.md
@@ -14,7 +14,7 @@ The models code is copied from the [transformers](https://github.com/huggingface
 We recommend creating a [conda environment](https://www.anaconda.com) following [BBT](https://github.com/txsun1997/Black-Box-Tuning#prepare-your-environment) 
 with the addition of installing NVFlare for running federated learning and some other updates:
 ```commandline
-conda create --name fedbpt python=3.8
+conda create --name fedbpt python=3.12
 conda activate fedbpt
 pip install -r requirements.txt
 ```
diff --git a/research/fed-bpt/requirements.txt b/research/fed-bpt/requirements.txt
index c840dde822..f2757c231f 100644
--- a/research/fed-bpt/requirements.txt
+++ b/research/fed-bpt/requirements.txt
@@ -1,8 +1,9 @@
 transformers==4.38.2
 fastNLP==0.6.0
 datasets
-cma
+cma==3.4.0
 scikit-learn
 tensorboard
 cvxopt
-nvflare~=2.4.1rc
\ No newline at end of file
+nvflare~=2.5.1rc
+