From 867f37d643e66c0223457c28f5345f2f21db97f2 Mon Sep 17 00:00:00 2001 From: Annie Tallund Date: Wed, 15 Mar 2023 11:27:08 +0100 Subject: Adapt rewrite module to MLIA coding standards - Fix imports - Update variable names - Refactor helper functions - Add licence headers - Add docstrings - Use f-strings rather than % notation - Create type annotations in rewrite module - Migrate from tqdm to rich progress bar - Use logging module in rewrite module: All print statements are replaced with logging module Resolves: MLIA-831, MLIA-842, MLIA-844, MLIA-846 Signed-off-by: Benjamin Klimczak Change-Id: Idee37538d72b9f01128a894281a8d10155f7c17c --- src/mlia/nn/rewrite/core/rewrite.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'src/mlia/nn/rewrite/core/rewrite.py') diff --git a/src/mlia/nn/rewrite/core/rewrite.py b/src/mlia/nn/rewrite/core/rewrite.py index d4f61c5..ab34b47 100644 --- a/src/mlia/nn/rewrite/core/rewrite.py +++ b/src/mlia/nn/rewrite/core/rewrite.py @@ -42,4 +42,5 @@ class Rewriter(Optimizer): return self.model def optimization_config(self) -> str: - """Optimization configirations.""" + """Optimization configurations.""" + return str(self.optimizer_configuration) -- cgit v1.2.1