Manage array init value from a scalar

whatsopt · Mar 19, 2023 · 1bfc635 · 1bfc635
1 parent efc8cf0
commit 1bfc635
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/app/lib/whats_opt/openmdao_variable.rb b/app/lib/whats_opt/openmdao_variable.rb
@@ -58,6 +58,10 @@ def default_py_value
       end
     end
 
+    def ones_py_value
+      default_py_value
+    end
+
     def lower_py_value
       "-inf"
     end

diff --git a/app/models/variable.rb b/app/models/variable.rb
@@ -94,7 +94,12 @@ def init_py_value
         default_py_value
       end
     else
-      self.parameter.init
+      is_num = true if Float(self.parameter.init) rescue false
+      if is_num && self.ndim > 0
+        ones_py_value + " * #{self.parameter.init}"
+      else
+        self.parameter.init
+      end
     end
   end