use_inf_as_na deprecated

materialsproject · Nov 27, 2023 · ffaeb22 · ffaeb22
1 parent 0e80835
commit ffaeb22
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mpcontribs-client/mpcontribs/client/__init__.py b/mpcontribs-client/mpcontribs/client/__init__.py
@@ -7,6 +7,7 @@
 import gzip
 import warnings
 import pandas as pd
+import numpy as np
 import plotly.io as pio
 import itertools
 import functools
@@ -88,7 +89,6 @@
 
 j2h = Json2Html()
 pd.options.plotting.backend = "plotly"
-pd.set_option('mode.use_inf_as_na', True)
 pio.templates.default = "simple_white"
 warnings.formatwarning = lambda msg, *args, **kwargs: f"{msg}\n"
 warnings.filterwarnings("default", category=DeprecationWarning, module=__name__)
@@ -383,6 +383,7 @@ def from_dict(cls, dct: dict):
 
     def _clean(self):
         """clean the dataframe"""
+        self.replace([np.inf, -np.inf], np.nan, inplace=True)
         self.fillna('', inplace=True)
         self.index = self.index.astype(str)
         for col in self.columns: