ASR: Support ListConstant in fill_dims_for_asr_type()

sunilgitb · Aug 28, 2023 · 5eab8a7 · 5eab8a7
1 parent 9b99b13
commit 5eab8a7
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/lpython/semantics/python_ast_to_asr.cpp b/src/lpython/semantics/python_ast_to_asr.cpp
@@ -1645,6 +1645,12 @@ class CommonVisitor : public AST::BaseVisitor<Struct> {
                 ASR::expr_t *value = tuple_constant->m_elements[i];
                 fill_dims_for_asr_type(dims, value, loc);
             }
+        } else if(ASR::is_a<ASR::ListConstant_t>(*value)) {
+            ASR::ListConstant_t* list_constant = ASR::down_cast<ASR::ListConstant_t>(value);
+            for( size_t i = 0; i < list_constant->n_args; i++ ) {
+                ASR::expr_t *value = list_constant->m_args[i];
+                fill_dims_for_asr_type(dims, value, loc);
+            }
         } else if(ASR::is_a<ASR::EnumValue_t>(*value)) {
             ASR::expr_t* enum_value = ASRUtils::expr_value(
                  ASR::down_cast<ASR::EnumValue_t>(value)->m_value);