Fix key error when checking for UTILITIES_UNIT_TESTING env var (#3563)

sonic-net · Oct 7, 2024 · 72d1faa · 72d1faa
1 parent 008a078
commit 72d1faa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pfc/main.py b/pfc/main.py
@@ -28,7 +28,7 @@ def dump_config_to_json(self, table_name, namespace):
         This function dumps the current config in a JSON file for unit testing.
         """
         # Only dump files in unit testing mode
-        if os.environ["UTILITIES_UNIT_TESTING"] != "2":
+        if os.getenv("UTILITIES_UNIT_TESTING") != "2":
             return
 
         if namespace not in self.updated_port_tables.keys():