annotated type for python containers

E3SM-Project · Oct 7, 2023 · ab592e8 · ab592e8
1 parent 1546f84
commit ab592e8
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/components/omega/create_scripts.py b/components/omega/create_scripts.py
@@ -9,6 +9,7 @@
 import re
 import stat
 import subprocess
+import typing
 import sys
 
 
@@ -140,7 +141,7 @@ def get_processed_value(self, raw_value, outvar):
     # get module info
     def get_modules(self, outvar):
 
-        modcmds = []
+        modcmds: typing.List[str] = []
         outvar["__OMEGA_MODULE_COMMANDS__"] = modcmds
 
         module_system_node = self.get_child("module_system")
@@ -212,7 +213,7 @@ def get_modules(self, outvar):
     # get environmental variables info
     def get_envs(self, outvar):
 
-        exports = {}
+        exports: typing.Dict[str, str] = {}
         outvar["__OMEGA_SCRIPT_EXPORTS__"] = exports
 
         envvar_nodes = self.get_children("environment_variables",
@@ -283,7 +284,7 @@ def get_mpirun(self, outvar):
     # collect machine info
     def gen_machinfo(self):
 
-        outvar = {}
+        outvar: typing.Dict[str, str] = {}
 
         self.get_mpirun(outvar)
         self.get_modules(outvar)