chore(frontend): add shortcut property to get server in fhe modules

zama-ai · Mar 29, 2024 · 4f48c25 · 4f48c25
1 parent 4068046
commit 4f48c25
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/frontends/concrete-python/concrete/fhe/compilation/module.py b/frontends/concrete-python/concrete/fhe/compilation/module.py
@@ -670,6 +670,13 @@ def statistics(self) -> Dict:
         }
         return statistics
 
+    @property
+    def server(self) -> Server:
+        """
+        Get the server used by the runtime.
+        """
+        return self.runtime.server
+
     def functions(self) -> Dict[str, FheFunction]:
         """
         Return a dictionnary containing all the functions of the module.

diff --git a/frontends/concrete-python/tests/compilation/test_program.py b/frontends/concrete-python/tests/compilation/test_program.py
@@ -71,10 +71,11 @@ def dec(x):
         composable=True,
         verbose=True,
     )
-    Module.compile(
+    compiled = Module.compile(
         {"inc": inputset, "dec": inputset},
         configuration,
     )
+    assert isinstance(compiled.server, fhe.Server)
 
 
 def test_compiled_clear_call(helpers):