codegen: Use python_codegen_method value, not its presence (#462)

Signed-off-by: Brad Keryan <[email protected]>
ni · Dec 14, 2023 · 847a244 · 847a244
1 parent 7f2849a
commit 847a244
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/codegen/utilities/function_helpers.py b/src/codegen/utilities/function_helpers.py
@@ -40,7 +40,7 @@ def get_functions(metadata, class_name=""):
     all_functions = deepcopy(metadata["functions"])
     functions_metadata = []
     for function_name, function_data in all_functions.items():
-        if "python_codegen_method" in function_data or function_name in EXCLUDED_FUNCTIONS:
+        if function_data.get("python_codegen_method") or function_name in EXCLUDED_FUNCTIONS:
             continue
         if (
             "python_class_name" in all_functions[function_name]