to_dict minor update

This commit is contained in:
Diogo Sousa
2023-09-27 13:03:34 +01:00
parent e66193d556
commit d2b3a0a2f3

View File

@@ -210,7 +210,8 @@ class OBBject(Tagged, Generic[T]):
df = self.to_dataframe().reset_index() # type: ignore
results = {}
for field in df.columns:
results[field] = df[field].tolist()
f = df[field].tolist()
results[field] = f[0] if len(f) == 1 else f
# remove index from results
if "index" in results: