From 262f9866f08a408cb474d3603749b49675699fc9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Florentin=20D=C3=B6rre?= <florentin.dorre@neo4j.com>
Date: Tue, 16 Jul 2024 09:26:17 +0200
Subject: [PATCH] Avoid warning about empty Series

diff between pandas 1 and 2. but we can actually just return the result if its empty
---
 graphdatascience/model/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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"] = {