Added error for classes that don't have an init function

Chillee · Chillee · commit f3d6ebe43191 · 2019-06-17T14:19:27.000-07:00
diff --git a/torch/csrc/jit/script/sugared_value.cpp b/torch/csrc/jit/script/sugared_value.cpp
@@ -248,6 +248,10 @@ std::shared_ptr<SugaredValue> ClassValue::call(
   // Generate a new object of the right type, then call `__init__` on it
   auto& g = *m.graph();
   auto self = g.insertNode(g.createObject(type_))->output();
+  if (!type_->hasAttribute("__init__")) {
+    throw ErrorReport(loc)
+        << "Class " << type_->basename() << " does not have an __init__ function defined.";
+  }
 
   // Call the init function
   MethodValue(self, "__init__").call(loc, m, inputs, attributes, n_binders);
