diff --git a/graphdatascience/model/model.py b/graphdatascience/model/model.py index 8d9d47ac4..9b1dadf78 100644 --- a/graphdatascience/model/model.py +++ b/graphdatascience/model/model.py @@ -201,7 +201,7 @@ def drop(self, failIfMissing: bool = False) -> Series[Any]: ).squeeze() if result.empty: - return Series() + return result # modelInfo {.*, modelName: modelName, modelType: modelType} AS modelInfo result["modelInfo"] = {