File tree Expand file tree Collapse file tree 1 file changed +5
-16
lines changed
sdks/python/apache_beam/internal Expand file tree Collapse file tree 1 file changed +5
-16
lines changed Original file line number Diff line number Diff line change @@ -236,24 +236,13 @@ def sample_func():
236236
237237 code_obj = sample_func .__code__
238238 original_filename = os .path .abspath (code_obj .co_filename )
239+ pickled_code = beam_cloudpickle .dumps (code_obj )
240+ unpickled_code = beam_cloudpickle .loads (pickled_code )
239241
240- try :
241- pickled_code = beam_cloudpickle .dumps (code_obj )
242- unpickled_code = beam_cloudpickle .loads (pickled_code )
242+ mock_filepath_interceptor .assert_called ()
243243
244- mock_filepath_interceptor .assert_called ()
245-
246- unpickled_filename = os .path .abspath (unpickled_code .co_filename )
247- self .assertEqual (unpickled_filename , original_filename )
248-
249- except AttributeError as e :
250- if 'get_code_object_params' in str (e ):
251- self .fail (
252- "Vendored cloudpickle BUG: AttributeError "
253- f"'get_code_object_params' raised during CodeType pickling. "
254- f"Error: { e } " )
255- else :
256- raise
244+ unpickled_filename = os .path .abspath (unpickled_code .co_filename )
245+ self .assertEqual (unpickled_filename , original_filename )
257246
258247
259248if __name__ == '__main__' :
You can’t perform that action at this time.
0 commit comments