Merge pull request #77 from jacquelinegarrahan/fix-keras-return

Fixes for array processing and updates to docs
slaclab · Oct 19, 2022 · a99aeae · a99aeae
2 parents 8ac285b + f625f57
commit a99aeae
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/docs/variables.md b/docs/variables.md
@@ -9,7 +9,6 @@
             - ImageOutputVariable
             - ArrayInputVariable
             - ArrayOutputVariable
-            - TableInputVariable
-            - TableOutputVariable
+            - TableVariable
     rendering:
         show_source: true
diff --git a/lume_model/utils.py b/lume_model/utils.py
@@ -127,12 +127,12 @@ def parse_variables(config: dict) -> Tuple[dict]:
                 if (
                     isinstance(variable_config["default"], (str,))
                     and value_type is not None
-                    and value_type == "str"
+                    and value_type == "string"
                 ):
                     pass
 
                 if isinstance(variable_config["default"], (str,)) and (
-                    value_type is None or value_type != "str"
+                    value_type is None or value_type != "string"
                 ):
                     variable_config["default"] = np.load(variable_config["default"])