tf-encrypted / moose

Secure distributed dataflow framework for encrypted machine learning and data processing
Apache License 2.0
57 stars 15 forks source link

PyMoose: remove need for explicit tracing in `LocalMooseRuntime` #1091

Closed mortendahl closed 2 years ago

mortendahl commented 2 years ago

Add the following to evaluation methods in LocalMooseRuntime and remove explicit calls to trace in examples etc:

if isinstance(computation, AbstractComputation):
    computation = trace(computation)