fix: updated pandas to allow for v2+

centre-for-humanities-computing · Oct 10, 2023 · 0b069e3 · 0b069e3
1 parent 5f092d6
commit 0b069e3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -220,6 +220,7 @@ commands =
   pytest -n auto {posargs:test}
 
 [testenv:type]
+allowlist_externals = pyright
 description: run type checks
 extras = tests, dev
 basepython = py39 # Setting these explicitly avoid recreating env if your shell is set to a different version

diff --git a/src/dacy/datasets/names.py b/src/dacy/datasets/names.py
@@ -51,6 +51,7 @@ def load_names(
         names = names.groupby(["name", "gender", "first_name"]).agg({"count": "sum"})
         # Change: groupby state_office and divide by sum
         names = names.groupby(level=0).apply(lambda x: x / float(x.sum().iloc[0]))
+        names = names.reset_index(level=0, drop=True)
         names = names.reset_index()
         names = names.loc[
             (names["gender"] == gender) & (names["count"] >= min_prop_gender)