github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-02-09 03:55:04 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat: add option to customize inputset length warning in eval on inputset function

Browse Source
This commit is contained in:
Umut
2022-03-03 10:43:52 +03:00
parent 3accaf5d96
commit f82096a79b
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
concrete/numpy/np_fhe_compiler.py
View File

@@ -171,12 +171,18 @@ class NPFHECompiler:
return None
return draw_graph(self._op_graph, show, vertical, save_to)
def eval_on_inputset(self, inputset: Union[Iterable[Any], Iterable[Tuple[Any, ...]]]) -> None:
def eval_on_inputset(
self,
inputset: Union[Iterable[Any], Iterable[Tuple[Any, ...]]],
warn_on_inputset_length: bool = False,
) -> None:
"""Evaluate the underlying function on an inputset in one go, populates OPGraph and bounds.
Args:
inputset (Union[Iterable[Any], Iterable[Tuple[Any, ...]]]): The inputset on which the
function should be evaluated.
warn_on_inputset_length (bool, optional): Set to True to get a warning
if inputset is not long enough. Defaults to False.
"""
inputset_iter = iter(inputset)
@@ -211,7 +217,7 @@ class NPFHECompiler:
self._compilation_configuration,
self.compilation_artifacts,
self._nodes_and_bounds,
False,
warn_on_inputset_length,
)
def _eval_on_current_inputset(self) -> None: