Fix certificates for federated_runtime.py

Signed-off-by: Ishant Thakare <[email protected]>
securefederatedai · Dec 16, 2024 · ec7248a · ec7248a
1 parent 79d989b
commit ec7248a
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 9 deletions.
diff --git a/...perimental/workflow/FederatedRuntime/101_MNIST/workspace/101_MNIST_FederatedRuntime.ipynb b/...perimental/workflow/FederatedRuntime/101_MNIST/workspace/101_MNIST_FederatedRuntime.ipynb
@@ -215,6 +215,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "#| export\n",
+    "\n",
     "import torchvision\n",
     "\n",
     "# Train and Test datasets\n",
@@ -600,9 +602,6 @@
     "director_info = {\n",
     "    'director_node_fqdn':'localhost',\n",
     "    'director_port':50050,\n",
-    "    'cert_chain': None,\n",
-    "    'api_cert': None,\n",
-    "    'api_private_key': None,\n",
     "}\n",
     "\n",
     "federated_runtime = FederatedRuntime(\n",

diff --git a/...mental/workflow/FederatedRuntime/301_MNIST_Watermaking/workspace/MNIST_Watermarking.ipynb b/...mental/workflow/FederatedRuntime/301_MNIST_Watermaking/workspace/MNIST_Watermarking.ipynb
@@ -461,9 +461,6 @@
     "director_info = {\n",
     "    'director_node_fqdn':'localhost',\n",
     "    'director_port':50050,\n",
-    "    'cert_chain': None,\n",
-    "    'api_cert': None,\n",
-    "    'api_private_key': None,\n",
     "}\n",
     "\n",
     "authorized_collaborators = ['Bangalore', 'Chandler']\n",

diff --git a/openfl/experimental/workflow/runtime/federated_runtime.py b/openfl/experimental/workflow/runtime/federated_runtime.py
@@ -60,9 +60,9 @@ def __init__(
         if director:
             self.director = director
             self._fill_certs(
-                self.director["cert_chain"],
-                self.director["api_private_key"],
-                self.director["api_cert"],
+                self.director.get("cert_chain", None),
+                self.director.get("api_private_key", None),
+                self.director.get("api_cert", None),
             )
             self._dir_client = self._create_director_client()