| from pathlib import Path | |
| import shutil | |
| from client_server_interface import FHEDev | |
| from matchers import AVAILABLE_MATCHERS, Matcher | |
| print("Generating deployment files for all available filters") | |
| # This repository's directory | |
| REPO_DIR = Path(__file__).parent | |
| # This repository's main necessary folders | |
| MATCHERS_PATH = REPO_DIR / "data/compiled_models" | |
| for matcher_name in AVAILABLE_MATCHERS: | |
| print("Matcher:", matcher_name, "\n") | |
| # Create the filter instance | |
| matcher = Matcher(matcher_name) | |
| # Compile the model on a representative inputset | |
| matcher.compile() | |
| # Define the directory path associated to this filter's deployment files | |
| deployment_path = MATCHERS_PATH / (matcher_name + "/deployment") | |
| # Delete the deployment folder and its content if it already exists | |
| if deployment_path.is_dir(): | |
| print("Delete the deployment folder and its content ...") | |
| shutil.rmtree(deployment_path) | |
| # Save the files needed for deployment | |
| print(f"Saving the files at {deployment_path} ...") | |
| fhe_dev_filter = FHEDev(matcher, deployment_path) | |
| fhe_dev_filter.save() | |
| print("Done !") | |