allow passing logical axis rules in cli

Author: jfacevedo-google

src/maxdiffusion/pyconfig.py

@@ -16,6 +16,7 @@
 
 # pylint: disable=missing-module-docstring
 import os
+import ast
 import json
 import sys
 from collections import OrderedDict
@@ -35,8 +36,10 @@ def string_to_bool(s: str) -> bool:
     return False
   raise ValueError(f"Can't convert {s} to bool")
 
+def string_to_list(string_list: str) -> list:
+  return ast.literal_eval(string_list)
 
-_yaml_types_to_parser = {str: str, int: int, float: float, bool: string_to_bool}
+_yaml_types_to_parser = {str: str, int: int, float: float, bool: string_to_bool, list: string_to_list}
 
 _config = None
 config = None