switch head(0) to use _meta directly

lincc-frameworks · Jul 16, 2024 · 5cf6320 · 5cf6320
1 parent ae39e3a
commit 5cf6320
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/nested_dask/utils/utils.py b/src/nested_dask/utils/utils.py
@@ -49,6 +49,6 @@ def count_nested(df, nested, by=None, join=True) -> NestedFrame:
     # and second depending on join
     if join:
         # adds the meta onto the existing meta
-        meta = pd.concat([df.head(0), meta])
+        meta = pd.concat([df._meta, meta])
 
     return df.map_partitions(lambda x: npd_utils.count_nested(x, nested, by=by, join=join), meta=meta)