コード例 #1
0
 def get_transforms(self) -> List[Dict[str, Any]]:
     """Get available transforms."""
     if self.config is None:
         raise ClientErrorException("Config not found.")
     framework = self.config.get("framework", None)
     if framework is None:
         raise ClientErrorException("Framework not set.")
     for fw_transforms in load_transforms_config():
         if fw_transforms.get("name") == framework:
             return fw_transforms.get("params", [])
     return []
コード例 #2
0
ファイル: params_feeder.py プロジェクト: intel/lpot
 def get_transforms(self) -> List[Dict[str, Any]]:
     """Get available transforms."""
     framework = self.config.get("framework", None)
     if framework is None:
         raise ClientErrorException("Framework not set.")
     domain = self.config.get("domain", None)
     transforms = []
     for fw_transforms in load_transforms_config():
         if fw_transforms.get("name") == framework:
             transforms = fw_transforms.get("params", [])
             break
     if domain is not None:
         transforms = filter_transforms(transforms, framework, domain)
     return transforms
コード例 #3
0
 def test_load_transforms_config(self) -> None:
     """Test getting transforms config."""
     result = load_transforms_config()
     self.assertIs(type(result), list)
     self.assertIsNot(result, [])