use rename_categories

yasa/hypno.py

@@ -514,7 +514,7 @@ def as_int(self):
         Name: Stage, dtype: int16
         """
         # Return as int16 (-32768 to 32767) to reduce memory usage
-        return self.hypno.map(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.