moiseenkov commented on code in PR #38531:
URL: https://github.com/apache/airflow/pull/38531#discussion_r1570868914
##########
airflow/operators/python.py:
##########
@@ -58,6 +56,15 @@
from airflow.utils.process_utils import execute_in_subprocess
from airflow.utils.python_virtualenv import prepare_virtualenv,
write_python_script
+log = logging.getLogger(__name__)
+
+if shutil.which("cloudpickle") or importlib.util.find_spec("cloudpickle"):
+ import cloudpickle as serialization_library
+elif shutil.which("dill") or importlib.util.find_spec("dill"):
+ import dill as serialization_library
+else:
+ log.warning("Neither dill and cloudpickle are installed. Please install
one with: pip install [name]")
Review Comment:
Fixed that. Thank you!
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
