ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-26 18:16:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add Gradio binding support

Browse Source
This commit is contained in:
allegroai
2023-02-28 17:12:11 +02:00
parent 22715cda19
commit 72b341ee51
3 changed files with 170 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
clearml/task.py
View File

@@ -74,6 +74,7 @@ from .binding.hydra_bind import PatchHydra
from .binding.click_bind import PatchClick
from .binding.fire_bind import PatchFire
from .binding.jsonargs_bind import PatchJsonArgParse
from .binding.gradio_bind import PatchGradio
from .binding.frameworks import WeightsFileHandler
from .config import (
config, DEV_TASK_NO_REUSE, get_is_master_node, DEBUG_SIMULATE_REMOTE_TASK, DEV_DEFAULT_OUTPUT_URI,
@@ -402,7 +403,7 @@ class Task(_Task):
'matplotlib': True, 'tensorflow': ['*.hdf5, 'something_else*], 'tensorboard': True,
'pytorch': ['*.pt'], 'xgboost': True, 'scikit': True, 'fastai': True,
'lightgbm': True, 'hydra': True, 'detect_repository': True, 'tfdefines': True,
'joblib': True, 'megengine': True, 'catboost': True
'joblib': True, 'megengine': True, 'catboost': True, 'gradio': True
}
.. code-block:: py
@@ -689,6 +690,8 @@ class Task(_Task):
PatchFastai.update_current_task(task)
if should_connect("lightgbm"):
PatchLIGHTgbmModelIO.update_current_task(task)
if should_connect("gradio"):
PatchGradio.update_current_task(task)
cls.__add_model_wildcards(auto_connect_frameworks)