Drop annotation of dynamic classes when pickling for Python < 3.7

CHANGES.md

@@ -3,6 +3,10 @@
 
 **This version requires Python 3.5 or later**
 
+- Stop pickling the annotations of a dynamic class for Python < 3.6
+  (follow up on #276)
+  ([issue #347](https://github.com/cloudpipe/cloudpickle/issues/347))
+
 1.3.0
 =====
 

cloudpickle/cloudpickle.py

@@ -581,6 +581,11 @@ def save_dynamic_class(self, obj):
         if isinstance(__dict__, property):
             type_kwargs['__dict__'] = __dict__
 
+        if sys.version_info < (3, 7):
+            # Although annotations were added in Python 3.4, It is not possible
+            # to properly pickle them until Python 3.7. (See #193)
+            clsdict.pop('__annotations__', None)
+
         save = self.save
         write = self.write
 

tests/cloudpickle_test.py

@@ -1802,6 +1802,73 @@ def g(x):
 
         self.assertEqual(f2.__annotations__, f.__annotations__)
 
+    @unittest.skipIf(sys.version_info >= (3, 7),
+                     "pickling annotations is supported starting Python 3.7")
+    def test_function_annotations_silent_dropping(self):
+        # Because of limitations of typing module, cloudpickle does not pickle
+        # the type annotations of a dynamic function or class for Python < 3.7
+
+        class UnpicklableAnnotation:
+            # Mock Annotation metaclass that errors out loudly if we try to
+            # pickle one of its instances
+            def __reduce__(self):
+                raise Exception("not picklable")
+
+        unpickleable_annotation = UnpicklableAnnotation()
+
+        def f(a: unpickleable_annotation):
+            return a
+
+        with pytest.raises(Exception):
+            cloudpickle.dumps(f.__annotations__)
+
+        depickled_f = pickle_depickle(f, protocol=self.protocol)
+        assert depickled_f.__annotations__ == {}
+
+    @unittest.skipIf(sys.version_info >= (3, 7) or sys.version_info < (3, 6),
+                     "pickling annotations is supported starting Python 3.7")
+    def test_class_annotations_silent_dropping(self):
+        # Because of limitations of typing module, cloudpickle does not pickle
+        # the type annotations of a dynamic function or class for Python < 3.7
+
+        # Pickling and unpickling must be done in different processes when
+        # testing dynamic classes (see #313)
+
+        code = '''if 1:
+        import cloudpickle
+        import sys
+
+        class UnpicklableAnnotation:
+            # Mock Annotation metaclass that errors out loudly if we try to
+            # pickle one of its instances
+            def __reduce__(self):
+                raise Exception("not picklable")
+
+        unpickleable_annotation = UnpicklableAnnotation()
+
+        class A:
+            a: unpickleable_annotation
+
+        try:
+            cloudpickle.dumps(A.__annotations__)
+        except Exception:
+            pass
+        else:
+            raise AssertionError
+
+        sys.stdout.buffer.write(cloudpickle.dumps(A, protocol={protocol}))
+        '''
+        cmd = [sys.executable, '-c', code.format(protocol=self.protocol)]
+        proc = subprocess.Popen(
+                cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE
+        )
+        proc.wait()
+        out, err = proc.communicate()
+        assert proc.returncode == 0, err
+
+        depickled_a = pickle.loads(out)
+        assert not hasattr(depickled_a, "__annotations__")
+
     @unittest.skipIf(sys.version_info < (3, 7),
                      """This syntax won't work on py2 and pickling annotations
                      isn't supported for py37 and below.""")