Updated deprecated functions (#167)

* Updated the now deprecated Series.replace() to Series.cat.rename_categories() Signed-off-by: S.J. Guillot <[email protected]> * Updated freq_str in plot_spectrogram() as 's' of pd.Timedelta is now preferred over 'S' Signed-off-by: S.J. Guillot <[email protected]> --------- Signed-off-by: S.J. Guillot <[email protected]>
raphaelvallat · Apr 7, 2024 · 5200696 · 5200696
1 parent c734753
commit 5200696
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/yasa/hypno.py b/yasa/hypno.py
@@ -482,7 +482,7 @@ def as_int(self):
         Name: Stage, dtype: int16
         """
         # Return as int16 (-32768 to 32767) to reduce memory usage
-        return self.hypno.replace(self.mapping).astype(np.int16)
+        return self.hypno.cat.rename_categories(self.mapping).astype(np.int16)
 
     def consolidate_stages(self, new_n_stages):
         """Reduce the number of stages in a hypnogram to match actigraphy or wearables.

diff --git a/yasa/plotting.py b/yasa/plotting.py
@@ -331,7 +331,7 @@ def plot_spectrogram(
 
     if hypno is not None:
         # Convert sampling frequency to pandas timefrequency string (e.g., "30s")
-        freq_str = pd.tseries.frequencies.to_offset(pd.Timedelta(1 / sf, "S")).freqstr
+        freq_str = pd.tseries.frequencies.to_offset(pd.Timedelta(1 / sf, "s")).freqstr
         # Create Hypnogram instance for plotting
         hyp = Hypnogram(hypno_int_to_str(hypno), freq=freq_str)
         hypnoplot_kwargs = dict(lw=1.5, fill_color=None)