actually fix test

lincc-frameworks · Aug 13, 2024 · 30e5859 · 30e5859
1 parent 5eef6a5
commit 30e5859
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/nested_pandas/nestedframe/core.py b/src/nested_pandas/nestedframe/core.py
@@ -275,7 +275,7 @@ def from_lists(cls, df, base_columns=None, list_columns=None, name="nested"):
             return df[base_columns].join(packed_df)
         # or just return the packed_df as a nestedframe if no base cols
         else:
-            return packed_df.to_frame()
+            return NestedFrame(packed_df.to_frame())
 
     def _split_query(self, expr) -> dict:
         """Splits a pandas query into multiple subqueries for nested and base layers"""

diff --git a/tests/nested_pandas/nestedframe/test_nestedframe.py b/tests/nested_pandas/nestedframe/test_nestedframe.py
@@ -352,7 +352,7 @@ def test_from_lists():
     assert list(res.columns) == ["nested"]
     assert list(res.nested_columns) == ["nested"]
 
-    res = NestedFrame.from_lists(nf[["e"]], base_columns=None, list_columns=["e"])
+    res = NestedFrame.from_lists(nf[["e"]], base_columns=None, list_columns=None)
     assert list(res.columns) == ["nested"]
     assert list(res.nested_columns) == ["nested"]