From 9c1c36886d4c2dc24069a403e4f2c6a0d4d615a1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Felipe=20Sodr=C3=A9?= <felipe.b4rros@gmail.com>
Date: Thu, 15 Feb 2024 19:39:21 -0300
Subject: [PATCH] update flatten: improve readability]

---
 crossfire/__init__.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/crossfire/__init__.py b/crossfire/__init__.py
index 5bbd55a..622790d 100644
--- a/crossfire/__init__.py
+++ b/crossfire/__init__.py
@@ -48,9 +48,9 @@ def flatten(data, nested_columns=None):
     nested_columns = set(nested_columns or NESTED_COLUMNS)
     if not nested_columns.issubset(NESTED_COLUMNS):
         raise NestedColumnError(nested_columns)
+    if not data:
+        return data
     if isinstance(data, list):
-        if not data:
-            return data
         keys = set(data[0].keys()) & nested_columns
         for item in data:
             for key in keys: