fix PR comments

sony · Jun 16, 2024 · c8d764c · c8d764c
1 parent 7a0830c
commit c8d764c
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 20 deletions.
diff --git a/model_compression_toolkit/xquant/keras/keras_report_utils.py b/model_compression_toolkit/xquant/keras/keras_report_utils.py
@@ -40,7 +40,7 @@ def __init__(self, report_dir: str):
         fw_impl = KerasImplementation()
 
         dataset_utils = KerasDatasetUtils()
-        model_folding = ModelFoldingUtils(fw_info=DEFAULT_KERAS_INFO,
+        model_folding = ModelFoldingUtils(fw_info=fw_info,
                                           fw_impl=fw_impl,
                                           fw_default_tpc=DEFAULT_KERAS_TPC)
 

diff --git a/tests/xquant_tests/pytorch_tests/test_model_analyzer.py b/tests/xquant_tests/pytorch_tests/test_model_analyzer.py
@@ -21,7 +21,7 @@
 from tests.xquant_tests.pytorch_tests.test_xquant_end2end import random_data_gen
 import model_compression_toolkit as mct
 from torch import nn
-class TestKerasModelAnalyzer(unittest.TestCase):
+class TestPytorchModelAnalyzer(unittest.TestCase):
 
     def setUp(self):
         self.device = get_working_device()

diff --git a/tutorials/notebooks/mct_features_notebooks/keras/example_keras_xquant.ipynb b/tutorials/notebooks/mct_features_notebooks/keras/example_keras_xquant.ipynb
diff --git a/tutorials/notebooks/mct_features_notebooks/pytorch/example_pytorch_xquant.ipynb b/tutorials/notebooks/mct_features_notebooks/pytorch/example_pytorch_xquant.ipynb