Merge pull request #728 from douglasjacobsen/software-formatting

Fix formatting of managed software
GoogleCloudPlatform · Oct 30, 2024 · 6b1431d · 6b1431d
2 parents 74d55b0 + 745857a
commit 6b1431d
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/lib/ramble/ramble/workspace/workspace.py b/lib/ramble/ramble/workspace/workspace.py
@@ -928,6 +928,10 @@ def manage_environments(
         software_dict = self.get_software_dict().copy()
 
         environments = software_dict[namespace.environments]
+        # Ensure package dict is an syaml_dict, for formatting
+        if not environments:
+            software_dict[namespace.environments] = syaml.syaml_dict()
+            environments = software_dict[namespace.environments]
 
         if remove:
             if env_name in environments:
@@ -996,6 +1000,11 @@ def manage_packages(
 
         packages = software_dict[namespace.packages]
 
+        # Ensure package dict is an syaml_dict, for formatting
+        if not packages:
+            software_dict[namespace.packages] = syaml.syaml_dict()
+            packages = software_dict[namespace.packages]
+
         if remove:
             if pkg_name in packages:
                 del packages[pkg_name]