Fixes ReadTheDocs build (#443)

Lightning-Universe · Jun 21, 2021 · 40fc879 · 40fc879
1 parent 41630a4
commit 40fc879
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/flash/core/model.py b/flash/core/model.py
@@ -91,9 +91,11 @@ def __new__(mcs, *args, **kwargs):
         result = ABCMeta.__new__(mcs, *args, **kwargs)
         if result.required_extras is not None:
             result.__init__ = _requires_extras(result.required_extras)(result.__init__)
-            result.load_from_checkpoint = classmethod(
-                _requires_extras(result.required_extras)(result.load_from_checkpoint.__func__)
-            )
+            load_from_checkpoint = getattr(result, "load_from_checkpoint", None)
+            if load_from_checkpoint is not None:
+                result.load_from_checkpoint = classmethod(
+                    _requires_extras(result.required_extras)(result.load_from_checkpoint.__func__)
+                )
         return result