Refine config check & add TF framework check (#1106)

Author: zehao-intel

neural_compressor/conf/config.py

@@ -1242,8 +1242,12 @@ def _read_cfg(self, cfg_fname):
                 with open(cfg_fname, 'w') as f:
                     f.write(content)
 
-            return validated_cfg
-
+            return validated_cfg   
+        except FileNotFoundError as f:
+            logger.error("{}.".format(f))
+            raise RuntimeError(
+                "The yaml file is not exist. Please check the file name or path."
+            )
         except Exception as e:
             logger.error("{}.".format(e))
             raise RuntimeError(

neural_compressor/experimental/component.py

@@ -79,6 +79,15 @@ def _init_with_conf(self):
         if self.cfg.model.framework != 'NA':
             self.framework = self.cfg.model.framework.lower()
             set_backend(self.framework)
+            if self.framework == 'tensorflow' or self.framework == 'inteltensorflow':
+                try:
+                    import tensorflow as tf
+                except Exception as e:
+                    logger.error("{}.".format(e))
+                    raise RuntimeError(
+                        "The TensorFlow framework is not correctly installed. Please check your environment"
+                    )
+
 
     def pre_process(self):
         """ Initialize the dataloader and train/eval functions from yaml config.

neural_compressor/model/model.py

@@ -179,7 +179,8 @@ def _is_mxnet(model):
         fwk_name = handler(model)
         if fwk_name != 'NA':
             break
-    assert fwk_name != 'NA', 'Framework is not detected correctly from model format.'
+    assert fwk_name != 'NA', 'Framework is not detected correctly from model format. This could be \
+caused by unsupported model or inappropriate framework installation.'
 
     return fwk_name
 

test/config/test_config.py

@@ -843,6 +843,12 @@ def test_data_type(self):
         transform_cfg = cfg['quantization']['calibration']['dataloader']['transform']['BilinearImagenet']
         self.assertTrue(isinstance(transform_cfg['mean_value'], list))
 
+    def test_yaml_detection(self):
+        try:
+            cfg = conf.Conf('not_exist.yaml').usr_cfg
+        except:
+            pass
+
     def test_deep_set(self):
       from neural_compressor.conf.dotdict import DotDict, deep_set
       cfg = {'evaluation': {'accuracy': {}}}
@@ -860,7 +866,6 @@ def test_deep_set(self):
       multi_metrics2 = dot_cfg['evaluation']['accuracy']['multi_metrics']
       self.assertTrue(multi_metrics1 == multi_metrics2)
       self.assertTrue(list(multi_metrics1.keys()) == ['weight', 'mAP'])
-      
 
 if __name__ == "__main__":
     unittest.main()