Added deepcopy to config_command (#505)

Author: nv-braf

model_analyzer/config/input/config_command.py

@@ -17,6 +17,8 @@
 import yaml
 from .yaml_config_validator import YamlConfigValidator
 
+from copy import deepcopy
+
 
 class ConfigCommand:
     """
@@ -167,5 +169,13 @@ def get_all_config(self):
 
         return config_dict
 
+    def __deepcopy__(self, memo):
+        cls = self.__class__
+        result = cls.__new__(cls)
+        memo[id(self)] = result
+        for k, v in self.__dict__.items():
+            setattr(result, k, deepcopy(v, memo))
+        return result
+
     def __getattr__(self, name):
         return self._fields[name].value()

tests/test_config.py

@@ -44,6 +44,8 @@
 from model_analyzer.constants import \
     CONFIG_PARSER_FAILURE
 
+from copy import deepcopy
+
 from unittest.mock import patch
 
 
@@ -1711,6 +1713,31 @@ def test_path_validation(self):
             self._evaluate_config(args, yaml_content, subcommand='profile')
         self.mock_os.set_os_path_exists_return_value(True)
 
+    def test_copy(self):
+        """
+        Test that deepcopy works correctly
+        """
+        args = [
+            'model-analyzer', 'profile', '--model-repository', 'cli_repository',
+            '-f', 'path-to-config-file', '--profile-models', 'vgg11'
+        ]
+        yaml_content = 'model_repository: yaml_repository'
+        configA = self._evaluate_config(args, yaml_content)
+
+        configB = deepcopy(configA)
+
+        self._assert_equality_of_model_configs(
+            configA.get_all_config()['profile_models'],
+            configB.get_all_config()['profile_models'])
+
+        self.assertEqual(configA.run_config_search_mode,
+                         configB.run_config_search_mode)
+
+        configB.run_config_search_mode = 'quick'
+
+        self.assertNotEqual(configA.run_config_search_mode,
+                            configB.run_config_search_mode)
+
 
 if __name__ == '__main__':
     unittest.main()