Remove support for PyPy (#8863)

jrbourbeau · web-flow · commit 7819dcc7a154 · 2022-03-31T01:51:22.000-05:00
diff --git a/dask/sizeof.py b/dask/sizeof.py
@@ -5,21 +5,12 @@
 
 from dask.utils import Dispatch
 
-try:  # PyPy does not support sys.getsizeof
-    sys.getsizeof(1)
-    getsizeof = sys.getsizeof
-except (AttributeError, TypeError):  # Monkey patch
-
-    def getsizeof(x):  # type: ignore
-        return 100
-
-
 sizeof = Dispatch(name="sizeof")
 
 
 @sizeof.register(object)
 def sizeof_default(o):
-    return getsizeof(o)
+    return sys.getsizeof(o)
 
 
 @sizeof.register(bytes)
@@ -53,9 +44,11 @@ def sizeof_python_collection(seq):
             samples = itertools.islice(seq, num_samples)
         else:
             samples = random.sample(seq, num_samples)
-        return getsizeof(seq) + int(num_items / num_samples * sum(map(sizeof, samples)))
+        return sys.getsizeof(seq) + int(
+            num_items / num_samples * sum(map(sizeof, samples))
+        )
     else:
-        return getsizeof(seq) + sum(map(sizeof, seq))
+        return sys.getsizeof(seq) + sum(map(sizeof, seq))
 
 
 class SimpleSizeof:
@@ -78,13 +71,13 @@ class SimpleSizeof:
 
 @sizeof.register(SimpleSizeof)
 def sizeof_blocked(d):
-    return getsizeof(d)
+    return sys.getsizeof(d)
 
 
 @sizeof.register(dict)
 def sizeof_python_dict(d):
     return (
-        getsizeof(d)
+        sys.getsizeof(d)
         + sizeof(list(d.keys()))
         + sizeof(list(d.values()))
         - 2 * sizeof(list())
diff --git a/dask/tests/test_sizeof.py b/dask/tests/test_sizeof.py
@@ -3,20 +3,20 @@
 
 import pytest
 
-from dask.sizeof import getsizeof, sizeof
+from dask.sizeof import sizeof
 from dask.utils import funcname
 
 
 def test_base():
-    assert sizeof(1) == getsizeof(1)
+    assert sizeof(1) == sys.getsizeof(1)
 
 
 def test_name():
     assert funcname(sizeof) == "sizeof"
 
 
 def test_containers():
-    assert sizeof([1, 2, [3]]) > (getsizeof(3) * 3 + getsizeof([]))
+    assert sizeof([1, 2, [3]]) > (sys.getsizeof(3) * 3 + sys.getsizeof([]))
 
 
 def test_bytes_like():
