Add a reset method to Config, fixes nv-morpheus#68

mdemoret-nv · Jan 26, 2022 · 9e1f213 · 9e1f213
1 parent 20c6b41
commit 9e1f213
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/morpheus/config.py b/morpheus/config.py
@@ -210,3 +210,9 @@ def to_string(self):
 
         # Using JSON serializer for now since its easier to read. pprint is more compact
         return json.dumps(dataclasses.asdict(self), indent=2, sort_keys=True)
+
+    @classmethod
+    def reset(cls):
+        cls.__default = None
+        cls.__instance = None
+        cls.__is_creating = False
diff --git a/tests/__init__.py b/tests/__init__.py
@@ -52,9 +52,8 @@ def setUp(self) -> None:
         self._mock_triton_servers_dir = MOCK_TRITON_DIR
 
     def tearDown(self) -> None:
-        # reset the config singleton work-around for #68
-        Config._Config__default = None
-        Config._Config__instance = None
+        # reset the config singleton
+        Config.reset()
 
         # Reset the asyncio event loop work-around for #69
         loop = asyncio.get_event_loop()