Thomas Chardonnens
commited on
Commit
•
4040d43
1
Parent(s):
9ed5c82
remove model_path from parameters
Browse files
client_server_interface.py
CHANGED
@@ -79,7 +79,7 @@ class FHEDev:
|
|
79 |
class FHEClient:
|
80 |
"""Client interface to encrypt and decrypt FHE data associated to a SeizureDetector."""
|
81 |
|
82 |
-
def __init__(self,
|
83 |
"""Initialize the FHE interface.
|
84 |
|
85 |
Args:
|
@@ -90,7 +90,7 @@ class FHEClient:
|
|
90 |
self.key_dir = key_dir
|
91 |
|
92 |
# If model_path does not exist raise
|
93 |
-
assert model_path.exists(), f"{model_path} does not exist. Please specify a valid path."
|
94 |
|
95 |
# Load the client
|
96 |
self.client = fhe.Client.load(self.model_path / "client.zip", self.key_dir)
|
|
|
79 |
class FHEClient:
|
80 |
"""Client interface to encrypt and decrypt FHE data associated to a SeizureDetector."""
|
81 |
|
82 |
+
def __init__(self, key_dir=None):
|
83 |
"""Initialize the FHE interface.
|
84 |
|
85 |
Args:
|
|
|
90 |
self.key_dir = key_dir
|
91 |
|
92 |
# If model_path does not exist raise
|
93 |
+
assert self.model_path.exists(), f"{self.model_path} does not exist. Please specify a valid path."
|
94 |
|
95 |
# Load the client
|
96 |
self.client = fhe.Client.load(self.model_path / "client.zip", self.key_dir)
|