Added error for classes that don't have an init function (#21880)

Chillee · facebook-github-bot · commit 5d67c606ea7d · 2019-06-19T21:33:37.000-07:00
Summary: Fixes #21761 Pull Request resolved: #21880 Differential Revision: D15879205 Pulled By: Chillee fbshipit-source-id: 8b614970196b381357b6032a73eeaab0b7a4f667
diff --git a/torch/csrc/jit/script/sugared_value.cpp b/torch/csrc/jit/script/sugared_value.cpp
@@ -271,6 +271,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_->getMethod("__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);
