akshaychitneni commented on code in PR #34840: URL: https://github.com/apache/airflow/pull/34840#discussion_r1385344445
########## airflow/providers/papermill/hooks/kernel.py: ########## @@ -0,0 +1,48 @@ +from airflow.hooks.base import BaseHook + +JUPYTER_KERNEL_SHELL_PORT = 60316 +JUPYTER_KERNEL_IOPUB_PORT = 60317 +JUPYTER_KERNEL_STDIN_PORT = 60318 +JUPYTER_KERNEL_CONTROL_PORT = 60319 +JUPYTER_KERNEL_HB_PORT = 60320 Review Comment: Added to docs. ########## airflow/providers/papermill/hooks/kernel.py: ########## @@ -0,0 +1,48 @@ +from airflow.hooks.base import BaseHook + +JUPYTER_KERNEL_SHELL_PORT = 60316 +JUPYTER_KERNEL_IOPUB_PORT = 60317 +JUPYTER_KERNEL_STDIN_PORT = 60318 +JUPYTER_KERNEL_CONTROL_PORT = 60319 +JUPYTER_KERNEL_HB_PORT = 60320 + + +def register_remote_engine(): + """ + Registers ``RemoteKernelEngine`` papermill engine + """ + from papermill.engines import papermill_engines + from airflow.providers.papermill.operators.papermill import RemoteKernelEngine, REMOTE_KERNEL_ENGINE + + papermill_engines.register(REMOTE_KERNEL_ENGINE, RemoteKernelEngine) + + +class KernelHook(BaseHook): + """ Review Comment: Added tests. -- 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: commits-unsubscr...@airflow.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org
