From 40fc87977537beec5b3ddaabaff4c1c3e0a9d63b Mon Sep 17 00:00:00 2001 From: Ethan Harris Date: Mon, 21 Jun 2021 19:43:15 +0100 Subject: [PATCH] Fixes ReadTheDocs build (#443) --- flash/core/model.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/flash/core/model.py b/flash/core/model.py index 20b72d08d1..0ad14d1c87 100644 --- 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