lib/torch/native/parser.rb in torch-rb-0.1.8 vs lib/torch/native/parser.rb in torch-rb-0.2.0

@@ -97,9 +97,11 @@
                 v.is_a?(Numeric)
               when "ScalarType?"
                 v.nil?
               when "bool"
                 v == true || v == false
+              when "str"
+                v.is_a?(String)
               else
                 raise Error, "Unknown argument type: #{arg_types[k]}. Please report a bug with #{@name}."
               end
 
             if !good