ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 13:23:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Patch Gradio root_path argument

Browse Source
This commit is contained in:
Alex Burlacu 2023-06-12 15:13:13 +03:00
parent be65b6055a
commit 4ab46dd92c
1 changed files with 43 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

69
clearml/binding/gradio_bind.py
View File

@ -3,6 +3,7 @@ from logging import getLogger
from .frameworks import _patched_call # noqa
from .import_bind import PostImportHookPatching
from ..utilities.networking import get_private_ip
from ..config import running_remotely
class PatchGradio:
@ -11,7 +12,7 @@ class PatchGradio:
_default_gradio_address = "0.0.0.0"
_default_gradio_port = 7860
_root_path_format = "/service/{}"
_root_path_format = "/service/{}/"
__server_config_warning = set()
@classmethod
@ -32,42 +33,55 @@ class PatchGradio:
try:
import gradio
gradio.networking.start_server = _patched_call(
gradio.networking.start_server, PatchGradio._patched_start_server
)
gradio.routes.App.__init__ = _patched_call(gradio.routes.App.__init__, PatchGradio._patched_init)
gradio.routes.App.get_blocks = _patched_call(gradio.routes.App.get_blocks, PatchGradio._patched_get_blocks)
gradio.blocks.Blocks.launch = _patched_call(gradio.blocks.Blocks.launch, PatchGradio._patched_launch)
except Exception:
pass
cls.__patched = True
@staticmethod
def _patched_start_server(original_fn, self, server_name=None, server_port=None, *args, **kwargs):
def _patched_get_blocks(original_fn, *args, **kwargs):
blocks = original_fn(*args, **kwargs)
if not PatchGradio._current_task or not running_remotely():
return blocks
blocks.config["root"] = PatchGradio._root_path_format.format(PatchGradio._current_task.id)
blocks.root = blocks.config["root"]
return blocks
@staticmethod
def _patched_launch(original_fn, *args, **kwargs):
if not PatchGradio._current_task:
return original_fn(self, server_name, server_port, *args, **kwargs)
return original_fn(*args, **kwargs)
PatchGradio.__warn_on_server_config(
kwargs.get("server_name"),
kwargs.get("server_port"),
kwargs.get("root_path")
)
if not running_remotely():
return original_fn(*args, **kwargs)
# noinspection PyProtectedMember
PatchGradio._current_task._set_runtime_properties(
{"_SERVICE": "EXTERNAL", "_ADDRESS": get_private_ip(), "_PORT": PatchGradio._default_gradio_port}
)
PatchGradio._current_task.set_system_tags(["external_service"])
PatchGradio.__warn_on_server_config(server_name, server_port)
server_name = PatchGradio._default_gradio_address
server_port = PatchGradio._default_gradio_port
return original_fn(self, server_name, server_port, *args, **kwargs)
@staticmethod
def _patched_init(original_fn, *args, **kwargs):
if not PatchGradio._current_task:
return original_fn(*args, **kwargs)
PatchGradio.__warn_on_server_config(kwargs.get("server_name"), kwargs.get("server_port"))
kwargs["root_path"] = PatchGradio._root_path_format.format(PatchGradio._current_task.id)
kwargs["root_path_in_servers"] = False
kwargs["server_name"] = PatchGradio._default_gradio_address
kwargs["server_port"] = PatchGradio._default_gradio_port
return original_fn(*args, **kwargs)
kwargs["root_path"] = PatchGradio._root_path_format.format(PatchGradio._current_task.id)
# noinspection PyBroadException
try:
return original_fn(*args, **kwargs)
except Exception as e:
del kwargs["root_path"]
return original_fn(*args, **kwargs)
@classmethod
def __warn_on_server_config(cls, server_name, server_port):
if server_name is None and server_port is None:
def __warn_on_server_config(cls, server_name, server_port, root_path):
if (server_name is None or server_name == PatchGradio._default_gradio_address) and \
(server_port is None and server_port == PatchGradio._default_gradio_port):
return
if (server_name, server_port, root_path) in cls.__server_config_warning:
return
cls.__server_config_warning.add((server_name, server_port, root_path))
if server_name is not None and server_port is not None:
server_config = "{}:{}".format(server_name, server_port)
what_to_ignore = "name and port"
@ -77,11 +91,14 @@ class PatchGradio:
else:
server_config = str(server_port)
what_to_ignore = "port"
if server_config in cls.__server_config_warning:
return
cls.__server_config_warning.add(server_config)
getLogger().warning(
"ClearML only supports '{}:{}'as the Gradio server. Ignoring {} '{}'".format(
"ClearML only supports '{}:{}' as the Gradio server. Ignoring {} '{}' in remote execution".format(
PatchGradio._default_gradio_address, PatchGradio._default_gradio_port, what_to_ignore, server_config
)
)
if root_path is not None:
getLogger().warning(
"ClearML will override root_path '{}' to '{}' in remote execution".format(
root_path, PatchGradio._root_path_format.format(PatchGradio._current_task.id)
)
)