Thomas Chardonnens commited on
Commit
4040d43
1 Parent(s): 9ed5c82

remove model_path from parameters

Browse files
Files changed (1) hide show
  1. client_server_interface.py +2 -2
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, model_path, key_dir=None):
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)