Skip to content

Commit

Permalink
Added handle for the workspace creation.
Browse files Browse the repository at this point in the history
  • Loading branch information
yhwen committed Nov 26, 2024
1 parent 1c15ae2 commit f424042
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion nvflare/private/fed/simulator/simulator_server.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import os
from typing import Dict, List, Optional

from nvflare.apis.fl_component import FLComponent
Expand Down Expand Up @@ -145,6 +145,8 @@ def _create_server_engine(self, args, snapshot_persistor):

def deploy(self, args, grpc_args=None, secure_train=False):
super(FederatedServer, self).deploy(args, grpc_args, secure_train)
os.makedirs(os.path.join(args.workspace, "local"), exist_ok=True)
os.makedirs(os.path.join(args.workspace, "startup"), exist_ok=True)
workspace = Workspace(args.workspace, "server", args.config_folder)
run_manager = RunManager(
server_name="server",
Expand Down

0 comments on commit f424042

Please sign in to comment.