Call DataFrame.map instead of DataFrame.applymap (#55)

Stonesjtu · Jul 16, 2024 · 212e595 · 212e595
1 parent a01086a
commit 212e595
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_memlab/line_profiler/line_records.py b/pytorch_memlab/line_profiler/line_records.py
@@ -181,7 +181,7 @@ def __repr__(self):
         for qual_name, merged in self._merge_line_records_with_code().items():
             maxlen = max(len(c) for c in merged.code)
             left_align = '{{:{maxlen}s}}'.format(maxlen=maxlen)
-            merged[byte_cols] = merged[byte_cols].applymap(readable_size)
+            merged[byte_cols] = merged[byte_cols].map(readable_size)
 
             # This is a mess, but I can't find any other way to left-align text strings.
             code_header = (left_align.format('code'), '', '')