handle warning

Seshat-Global-History-Databank · Oct 29, 2024 · 37692bb · 37692bb
1 parent 1fcc8d1
commit 37692bb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/camels.ipynb b/examples/camels.ipynb
@@ -115,7 +115,7 @@
    "outputs": [],
    "source": [
     "# Create a new empty DataFrame to store the frequency of each camel value for each year\n",
-    "frequency_df = pd.DataFrame(index=years, columns=absent_present_values).fillna(0)\n",
+    "frequency_df = pd.DataFrame(index=years, columns=absent_present_values).infer_objects(copy=False).fillna(0)\n",
     "print(frequency_df.sample(5))  # Show a random sample of 5 rows"
    ]
   },

diff --git a/seshat_api/seshat_api.py b/seshat_api/seshat_api.py
@@ -297,7 +297,7 @@ def get_frequencies(client, class_names, years, value='present'):
         dataframes.append(polities_with_var_df)
 
     # Create a DataFrame to store the frequency of each variable having the value across polities over time
-    frequency_df = pd.DataFrame(index=years, columns=variables).fillna(0)
+    frequency_df = pd.DataFrame(index=years, columns=variables).infer_objects(copy=False).fillna(0)
 
     # Count the number of polities with the value for each variable for each year
     for year in years: