mirror of
https://github.com/gradio-app/gradio.git
synced 2025-04-06 12:30:29 +08:00
Simplify tests (#3608)
* simplify tests * imports * imports * formatting * removed cometml typing * simplify * changelog
This commit is contained in:
Abubakar Abid
GitHub
parent
811d26aab4
commit
1c8d3cb05d
@ -20,7 +20,7 @@
|
||||
|
||||
## Testing and Infrastructure Changes:
|
||||
|
||||
No changes to highlight.
|
||||
- Removed heavily-mocked tests related to comet_ml, wandb, and mlflow as they added a significant amount of test dependencies that prevented installation of test dependencies on Windows environemnts. By [@abidlabs](https://github.com/abidlabs) in [PR 3608](https://github.com/gradio-app/gradio/pull/3608)
|
||||
|
||||
## Breaking Changes:
|
||||
|
||||
|
||||
@ -51,7 +51,6 @@ from gradio.utils import (
|
||||
set_documentation_group("blocks")
|
||||
|
||||
if TYPE_CHECKING: # Only import for type checking (is False at runtime).
|
||||
import comet_ml
|
||||
from fastapi.applications import FastAPI
|
||||
|
||||
from gradio.components import Component
|
||||
@ -1688,7 +1687,7 @@ class Blocks(BlockContext):
|
||||
|
||||
def integrate(
|
||||
self,
|
||||
comet_ml: comet_ml.Experiment | None = None,
|
||||
comet_ml=None,
|
||||
wandb: ModuleType | None = None,
|
||||
mlflow: ModuleType | None = None,
|
||||
) -> None:
|
||||
|
||||
13
scripts/create_test_requirements-37.sh
Normal file
13
scripts/create_test_requirements-37.sh
Normal file
@ -0,0 +1,13 @@
|
||||
#!/bin/bash
|
||||
|
||||
cd "$(dirname ${0})/.."
|
||||
source scripts/helpers.sh
|
||||
|
||||
pip_required
|
||||
|
||||
echo "Creating test requirements...
|
||||
To match the CI environment, this script should be run from a Unix-like system in Python 3.7."
|
||||
|
||||
cd test
|
||||
pip install --upgrade pip-tools
|
||||
pip-compile requirements.in --output-file requirements-37.txt
|
||||
@ -6,7 +6,8 @@ source scripts/helpers.sh
|
||||
pip_required
|
||||
|
||||
echo "Creating test requirements...
|
||||
It's recommended to run this script from a Unix-like system in Python 3.7 (or higher) environment."
|
||||
To match the CI environment, this script should be run from a Unix-like system in Python 3.9."
|
||||
|
||||
cd test
|
||||
pip install --upgrade pip-tools
|
||||
pip-compile --output-file requirements.txt
|
||||
pip-compile requirements.in --output-file requirements.txt
|
||||
|
||||
@ -2,18 +2,14 @@
|
||||
# This file is autogenerated by pip-compile with Python 3.7
|
||||
# by the following command:
|
||||
#
|
||||
# pip-compile --output-file=requirements.txt
|
||||
# pip-compile --output-file=requirements-37.txt requirements.in
|
||||
#
|
||||
alembic==1.8.1
|
||||
# via mlflow
|
||||
altair==4.2.0
|
||||
# via -r requirements.in
|
||||
anyio==3.6.1
|
||||
# via
|
||||
# httpcore
|
||||
# starlette
|
||||
appnope==0.1.3
|
||||
# via ipython
|
||||
asyncio==3.4.3
|
||||
# via -r requirements.in
|
||||
attrs==21.4.0
|
||||
@ -32,48 +28,23 @@ botocore==1.29.87
|
||||
# s3transfer
|
||||
certifi==2022.6.15
|
||||
# via
|
||||
# dulwich
|
||||
# httpcore
|
||||
# httpx
|
||||
# requests
|
||||
# sentry-sdk
|
||||
charset-normalizer==2.1.0
|
||||
# via requests
|
||||
click==8.1.3
|
||||
# via
|
||||
# black
|
||||
# databricks-cli
|
||||
# flask
|
||||
# mlflow
|
||||
# wandb
|
||||
# via black
|
||||
cloudpickle==2.1.0
|
||||
# via
|
||||
# mlflow
|
||||
# shap
|
||||
comet-ml==3.31.6
|
||||
# via -r requirements.in
|
||||
configobj==5.0.6
|
||||
# via everett
|
||||
# via shap
|
||||
coverage[toml]==6.4.2
|
||||
# via
|
||||
# -r requirements.in
|
||||
# pytest-cov
|
||||
databricks-cli==0.17.0
|
||||
# via mlflow
|
||||
decorator==5.1.1
|
||||
# via ipython
|
||||
docker==5.0.3
|
||||
# via mlflow
|
||||
docker-pycreds==0.4.0
|
||||
# via wandb
|
||||
dulwich==0.20.45
|
||||
# via comet-ml
|
||||
entrypoints==0.4
|
||||
# via
|
||||
# altair
|
||||
# mlflow
|
||||
everett[ini]==3.0.0
|
||||
# via comet-ml
|
||||
# via altair
|
||||
fastapi==0.87.0
|
||||
# via -r requirements.in
|
||||
filelock==3.7.1
|
||||
@ -82,18 +53,6 @@ filelock==3.7.1
|
||||
# transformers
|
||||
flake8==4.0.1
|
||||
# via -r requirements.in
|
||||
flask==2.1.3
|
||||
# via
|
||||
# mlflow
|
||||
# prometheus-flask-exporter
|
||||
gitdb==4.0.9
|
||||
# via gitpython
|
||||
gitpython==3.1.27
|
||||
# via
|
||||
# mlflow
|
||||
# wandb
|
||||
gunicorn==20.1.0
|
||||
# via mlflow
|
||||
h11==0.12.0
|
||||
# via httpcore
|
||||
httpcore==0.15.0
|
||||
@ -115,22 +74,25 @@ imageio==2.19.5
|
||||
# via scikit-image
|
||||
importlib-metadata==4.2.0
|
||||
# via
|
||||
# flask
|
||||
# mlflow
|
||||
# click
|
||||
# flake8
|
||||
# huggingface-hub
|
||||
# jsonschema
|
||||
# pluggy
|
||||
# pytest
|
||||
# transformers
|
||||
importlib-resources==5.12.0
|
||||
# via jsonschema
|
||||
iniconfig==1.1.1
|
||||
# via pytest
|
||||
ipython==7.34.0
|
||||
# via -r requirements.in
|
||||
isort==5.10.1
|
||||
# via -r requirements.in
|
||||
itsdangerous==2.1.2
|
||||
# via flask
|
||||
jedi==0.18.1
|
||||
# via ipython
|
||||
jinja2==3.1.2
|
||||
# via
|
||||
# altair
|
||||
# flask
|
||||
# via altair
|
||||
jmespath==1.0.1
|
||||
# via
|
||||
# boto3
|
||||
@ -138,23 +100,15 @@ jmespath==1.0.1
|
||||
joblib==1.1.0
|
||||
# via scikit-learn
|
||||
jsonschema==4.7.2
|
||||
# via
|
||||
# altair
|
||||
# comet-ml
|
||||
# via altair
|
||||
llvmlite==0.38.1
|
||||
# via numba
|
||||
mako==1.2.1
|
||||
# via alembic
|
||||
markupsafe==2.1.1
|
||||
# via
|
||||
# jinja2
|
||||
# mako
|
||||
# via jinja2
|
||||
matplotlib-inline==0.1.3
|
||||
# via ipython
|
||||
mccabe==0.6.1
|
||||
# via flake8
|
||||
mlflow==1.27.0
|
||||
# via -r requirements.in
|
||||
mypy-extensions==0.4.3
|
||||
# via black
|
||||
networkx==2.6.3
|
||||
@ -165,7 +119,6 @@ numpy==1.21.6
|
||||
# via
|
||||
# altair
|
||||
# imageio
|
||||
# mlflow
|
||||
# numba
|
||||
# pandas
|
||||
# pywavelets
|
||||
@ -175,14 +128,9 @@ numpy==1.21.6
|
||||
# shap
|
||||
# tifffile
|
||||
# transformers
|
||||
nvidia-ml-py3==7.352.0
|
||||
# via comet-ml
|
||||
oauthlib==3.2.0
|
||||
# via databricks-cli
|
||||
packaging==21.3
|
||||
# via
|
||||
# huggingface-hub
|
||||
# mlflow
|
||||
# pytest
|
||||
# scikit-image
|
||||
# shap
|
||||
@ -190,15 +138,12 @@ packaging==21.3
|
||||
pandas==1.3.5
|
||||
# via
|
||||
# altair
|
||||
# mlflow
|
||||
# shap
|
||||
# vega-datasets
|
||||
parso==0.8.3
|
||||
# via jedi
|
||||
pathspec==0.9.0
|
||||
# via black
|
||||
pathtools==0.1.2
|
||||
# via wandb
|
||||
pexpect==4.8.0
|
||||
# via ipython
|
||||
pickleshare==0.7.5
|
||||
@ -211,20 +156,8 @@ platformdirs==2.5.2
|
||||
# via black
|
||||
pluggy==1.0.0
|
||||
# via pytest
|
||||
prometheus-client==0.14.1
|
||||
# via prometheus-flask-exporter
|
||||
prometheus-flask-exporter==0.20.2
|
||||
# via mlflow
|
||||
promise==2.3
|
||||
# via wandb
|
||||
prompt-toolkit==3.0.30
|
||||
# via ipython
|
||||
protobuf==3.20.1
|
||||
# via
|
||||
# mlflow
|
||||
# wandb
|
||||
psutil==5.9.1
|
||||
# via wandb
|
||||
ptyprocess==0.7.0
|
||||
# via pexpect
|
||||
py==1.11.0
|
||||
@ -239,8 +172,6 @@ pyflakes==2.4.0
|
||||
# via flake8
|
||||
pygments==2.12.0
|
||||
# via ipython
|
||||
pyjwt==2.4.0
|
||||
# via databricks-cli
|
||||
pyparsing==3.0.9
|
||||
# via packaging
|
||||
pyrsistent==0.18.1
|
||||
@ -259,33 +190,19 @@ python-dateutil==2.8.2
|
||||
# botocore
|
||||
# pandas
|
||||
pytz==2022.1
|
||||
# via
|
||||
# mlflow
|
||||
# pandas
|
||||
# via pandas
|
||||
pywavelets==1.3.0
|
||||
# via scikit-image
|
||||
pyyaml==6.0
|
||||
# via
|
||||
# huggingface-hub
|
||||
# mlflow
|
||||
# transformers
|
||||
# wandb
|
||||
querystring-parser==1.2.4
|
||||
# via mlflow
|
||||
regex==2022.7.9
|
||||
# via transformers
|
||||
requests==2.28.1
|
||||
# via
|
||||
# comet-ml
|
||||
# databricks-cli
|
||||
# docker
|
||||
# huggingface-hub
|
||||
# mlflow
|
||||
# requests-toolbelt
|
||||
# transformers
|
||||
# wandb
|
||||
requests-toolbelt==0.9.1
|
||||
# via comet-ml
|
||||
respx==0.19.2
|
||||
# via -r requirements.in
|
||||
rfc3986[idna2008]==1.5.0
|
||||
@ -298,51 +215,22 @@ scikit-learn==1.0.2
|
||||
# via shap
|
||||
scipy==1.7.3
|
||||
# via
|
||||
# mlflow
|
||||
# scikit-image
|
||||
# scikit-learn
|
||||
# shap
|
||||
semantic-version==2.10.0
|
||||
# via comet-ml
|
||||
sentry-sdk==1.7.2
|
||||
# via
|
||||
# comet-ml
|
||||
# wandb
|
||||
setproctitle==1.2.3
|
||||
# via wandb
|
||||
shap==0.41.0
|
||||
# via -r requirements.in
|
||||
shortuuid==1.0.9
|
||||
# via wandb
|
||||
six==1.16.0
|
||||
# via
|
||||
# comet-ml
|
||||
# configobj
|
||||
# databricks-cli
|
||||
# docker-pycreds
|
||||
# promise
|
||||
# python-dateutil
|
||||
# querystring-parser
|
||||
# wandb
|
||||
# via python-dateutil
|
||||
slicer==0.0.7
|
||||
# via shap
|
||||
smmap==5.0.0
|
||||
# via gitdb
|
||||
sniffio==1.2.0
|
||||
# via
|
||||
# anyio
|
||||
# httpcore
|
||||
# httpx
|
||||
sqlalchemy==1.4.39
|
||||
# via
|
||||
# alembic
|
||||
# mlflow
|
||||
sqlparse==0.4.2
|
||||
# via mlflow
|
||||
starlette==0.21.0
|
||||
# via fastapi
|
||||
tabulate==0.8.10
|
||||
# via databricks-cli
|
||||
threadpoolctl==3.1.0
|
||||
# via scikit-learn
|
||||
tifffile==2021.11.2
|
||||
@ -370,37 +258,31 @@ traitlets==5.3.0
|
||||
# matplotlib-inline
|
||||
transformers==4.20.1
|
||||
# via -r requirements.in
|
||||
typed-ast==1.5.4
|
||||
# via black
|
||||
typing-extensions==4.3.0
|
||||
# via
|
||||
# anyio
|
||||
# black
|
||||
# huggingface-hub
|
||||
# importlib-metadata
|
||||
# jsonschema
|
||||
# pydantic
|
||||
# pytest-asyncio
|
||||
# starlette
|
||||
# torch
|
||||
urllib3==1.26.10
|
||||
# via
|
||||
# botocore
|
||||
# dulwich
|
||||
# requests
|
||||
# sentry-sdk
|
||||
vega-datasets==0.9.0
|
||||
# via -r requirements.in
|
||||
wandb==0.12.21
|
||||
# via -r requirements.in
|
||||
wcwidth==0.2.5
|
||||
# via prompt-toolkit
|
||||
websocket-client==1.3.3
|
||||
# via
|
||||
# comet-ml
|
||||
# docker
|
||||
werkzeug==2.1.2
|
||||
# via flask
|
||||
wrapt==1.14.1
|
||||
# via comet-ml
|
||||
wurlitzer==3.0.2
|
||||
# via comet-ml
|
||||
zipp==3.8.1
|
||||
# via importlib-metadata
|
||||
# via
|
||||
# importlib-metadata
|
||||
# importlib-resources
|
||||
|
||||
# The following packages are considered to be unsafe in a requirements file:
|
||||
# setuptools
|
||||
|
||||
@ -1,16 +1,13 @@
|
||||
# Don't forget to run bash scripts/create_test_requirements.sh from unix or wsl when you update this file.
|
||||
# Don't forget to run bash scripts/create_test_requirements.sh and scripts/create_test_requirements-37.sh from unix or wsl when you update this file.
|
||||
asyncio
|
||||
boto3
|
||||
IPython
|
||||
comet_ml
|
||||
coverage
|
||||
mlflow
|
||||
torch
|
||||
transformers
|
||||
scikit-image
|
||||
shap
|
||||
pytest
|
||||
wandb
|
||||
huggingface_hub
|
||||
pytest-cov
|
||||
pytest-asyncio
|
||||
|
||||
@ -2,18 +2,14 @@
|
||||
# This file is autogenerated by pip-compile with Python 3.9
|
||||
# by the following command:
|
||||
#
|
||||
# pip-compile --output-file=requirements.txt
|
||||
# pip-compile --output-file=requirements.txt requirements.in
|
||||
#
|
||||
alembic==1.8.1
|
||||
# via mlflow
|
||||
altair==4.2.0
|
||||
# via -r requirements.in
|
||||
anyio==3.6.1
|
||||
# via
|
||||
# httpcore
|
||||
# starlette
|
||||
appnope==0.1.3
|
||||
# via ipython
|
||||
asyncio==3.4.3
|
||||
# via -r requirements.in
|
||||
attrs==21.4.0
|
||||
@ -32,48 +28,23 @@ botocore==1.29.87
|
||||
# s3transfer
|
||||
certifi==2022.6.15
|
||||
# via
|
||||
# dulwich
|
||||
# httpcore
|
||||
# httpx
|
||||
# requests
|
||||
# sentry-sdk
|
||||
charset-normalizer==2.1.0
|
||||
# via requests
|
||||
click==8.1.3
|
||||
# via
|
||||
# black
|
||||
# databricks-cli
|
||||
# flask
|
||||
# mlflow
|
||||
# wandb
|
||||
# via black
|
||||
cloudpickle==2.1.0
|
||||
# via
|
||||
# mlflow
|
||||
# shap
|
||||
comet-ml==3.31.6
|
||||
# via -r requirements.in
|
||||
configobj==5.0.6
|
||||
# via everett
|
||||
# via shap
|
||||
coverage[toml]==6.4.2
|
||||
# via
|
||||
# -r requirements.in
|
||||
# pytest-cov
|
||||
databricks-cli==0.17.0
|
||||
# via mlflow
|
||||
decorator==5.1.1
|
||||
# via ipython
|
||||
docker==5.0.3
|
||||
# via mlflow
|
||||
docker-pycreds==0.4.0
|
||||
# via wandb
|
||||
dulwich==0.20.45
|
||||
# via comet-ml
|
||||
entrypoints==0.4
|
||||
# via
|
||||
# altair
|
||||
# mlflow
|
||||
everett[ini]==3.0.0
|
||||
# via comet-ml
|
||||
# via altair
|
||||
fastapi==0.87.0
|
||||
# via -r requirements.in
|
||||
filelock==3.7.1
|
||||
@ -82,18 +53,6 @@ filelock==3.7.1
|
||||
# transformers
|
||||
flake8==4.0.1
|
||||
# via -r requirements.in
|
||||
flask==2.1.3
|
||||
# via
|
||||
# mlflow
|
||||
# prometheus-flask-exporter
|
||||
gitdb==4.0.9
|
||||
# via gitpython
|
||||
gitpython==3.1.27
|
||||
# via
|
||||
# mlflow
|
||||
# wandb
|
||||
gunicorn==20.1.0
|
||||
# via mlflow
|
||||
h11==0.12.0
|
||||
# via httpcore
|
||||
httpcore==0.15.0
|
||||
@ -113,24 +72,16 @@ idna==3.3
|
||||
# rfc3986
|
||||
imageio==2.19.5
|
||||
# via scikit-image
|
||||
importlib-metadata==4.2.0
|
||||
# via
|
||||
# flask
|
||||
# mlflow
|
||||
iniconfig==1.1.1
|
||||
# via pytest
|
||||
ipython==7.34.0
|
||||
# via -r requirements.in
|
||||
isort==5.10.1
|
||||
# via -r requirements.in
|
||||
itsdangerous==2.1.2
|
||||
# via flask
|
||||
jedi==0.18.1
|
||||
# via ipython
|
||||
jinja2==3.1.2
|
||||
# via
|
||||
# altair
|
||||
# flask
|
||||
# via altair
|
||||
jmespath==1.0.1
|
||||
# via
|
||||
# boto3
|
||||
@ -138,23 +89,15 @@ jmespath==1.0.1
|
||||
joblib==1.1.0
|
||||
# via scikit-learn
|
||||
jsonschema==4.7.2
|
||||
# via
|
||||
# altair
|
||||
# comet-ml
|
||||
# via altair
|
||||
llvmlite==0.38.1
|
||||
# via numba
|
||||
mako==1.2.1
|
||||
# via alembic
|
||||
markupsafe==2.1.1
|
||||
# via
|
||||
# jinja2
|
||||
# mako
|
||||
# via jinja2
|
||||
matplotlib-inline==0.1.3
|
||||
# via ipython
|
||||
mccabe==0.6.1
|
||||
# via flake8
|
||||
mlflow==1.27.0
|
||||
# via -r requirements.in
|
||||
mypy-extensions==0.4.3
|
||||
# via black
|
||||
networkx==2.6.3
|
||||
@ -165,7 +108,6 @@ numpy==1.21.6
|
||||
# via
|
||||
# altair
|
||||
# imageio
|
||||
# mlflow
|
||||
# numba
|
||||
# pandas
|
||||
# pywavelets
|
||||
@ -175,14 +117,9 @@ numpy==1.21.6
|
||||
# shap
|
||||
# tifffile
|
||||
# transformers
|
||||
nvidia-ml-py3==7.352.0
|
||||
# via comet-ml
|
||||
oauthlib==3.2.0
|
||||
# via databricks-cli
|
||||
packaging==21.3
|
||||
# via
|
||||
# huggingface-hub
|
||||
# mlflow
|
||||
# pytest
|
||||
# scikit-image
|
||||
# shap
|
||||
@ -190,15 +127,12 @@ packaging==21.3
|
||||
pandas==1.3.5
|
||||
# via
|
||||
# altair
|
||||
# mlflow
|
||||
# shap
|
||||
# vega-datasets
|
||||
parso==0.8.3
|
||||
# via jedi
|
||||
pathspec==0.9.0
|
||||
# via black
|
||||
pathtools==0.1.2
|
||||
# via wandb
|
||||
pexpect==4.8.0
|
||||
# via ipython
|
||||
pickleshare==0.7.5
|
||||
@ -211,20 +145,8 @@ platformdirs==2.5.2
|
||||
# via black
|
||||
pluggy==1.0.0
|
||||
# via pytest
|
||||
prometheus-client==0.14.1
|
||||
# via prometheus-flask-exporter
|
||||
prometheus-flask-exporter==0.20.2
|
||||
# via mlflow
|
||||
promise==2.3
|
||||
# via wandb
|
||||
prompt-toolkit==3.0.30
|
||||
# via ipython
|
||||
protobuf==3.20.1
|
||||
# via
|
||||
# mlflow
|
||||
# wandb
|
||||
psutil==5.9.1
|
||||
# via wandb
|
||||
ptyprocess==0.7.0
|
||||
# via pexpect
|
||||
py==1.11.0
|
||||
@ -239,8 +161,6 @@ pyflakes==2.4.0
|
||||
# via flake8
|
||||
pygments==2.12.0
|
||||
# via ipython
|
||||
pyjwt==2.4.0
|
||||
# via databricks-cli
|
||||
pyparsing==3.0.9
|
||||
# via packaging
|
||||
pyrsistent==0.18.1
|
||||
@ -259,33 +179,19 @@ python-dateutil==2.8.2
|
||||
# botocore
|
||||
# pandas
|
||||
pytz==2022.1
|
||||
# via
|
||||
# mlflow
|
||||
# pandas
|
||||
# via pandas
|
||||
pywavelets==1.3.0
|
||||
# via scikit-image
|
||||
pyyaml==6.0
|
||||
# via
|
||||
# huggingface-hub
|
||||
# mlflow
|
||||
# transformers
|
||||
# wandb
|
||||
querystring-parser==1.2.4
|
||||
# via mlflow
|
||||
regex==2022.7.9
|
||||
# via transformers
|
||||
requests==2.28.1
|
||||
# via
|
||||
# comet-ml
|
||||
# databricks-cli
|
||||
# docker
|
||||
# huggingface-hub
|
||||
# mlflow
|
||||
# requests-toolbelt
|
||||
# transformers
|
||||
# wandb
|
||||
requests-toolbelt==0.9.1
|
||||
# via comet-ml
|
||||
respx==0.19.2
|
||||
# via -r requirements.in
|
||||
rfc3986[idna2008]==1.5.0
|
||||
@ -298,51 +204,22 @@ scikit-learn==1.0.2
|
||||
# via shap
|
||||
scipy==1.7.3
|
||||
# via
|
||||
# mlflow
|
||||
# scikit-image
|
||||
# scikit-learn
|
||||
# shap
|
||||
semantic-version==2.10.0
|
||||
# via comet-ml
|
||||
sentry-sdk==1.7.2
|
||||
# via
|
||||
# comet-ml
|
||||
# wandb
|
||||
setproctitle==1.2.3
|
||||
# via wandb
|
||||
shap==0.41.0
|
||||
# via -r requirements.in
|
||||
shortuuid==1.0.9
|
||||
# via wandb
|
||||
six==1.16.0
|
||||
# via
|
||||
# comet-ml
|
||||
# configobj
|
||||
# databricks-cli
|
||||
# docker-pycreds
|
||||
# promise
|
||||
# python-dateutil
|
||||
# querystring-parser
|
||||
# wandb
|
||||
# via python-dateutil
|
||||
slicer==0.0.7
|
||||
# via shap
|
||||
smmap==5.0.0
|
||||
# via gitdb
|
||||
sniffio==1.2.0
|
||||
# via
|
||||
# anyio
|
||||
# httpcore
|
||||
# httpx
|
||||
sqlalchemy==1.4.39
|
||||
# via
|
||||
# alembic
|
||||
# mlflow
|
||||
sqlparse==0.4.2
|
||||
# via mlflow
|
||||
starlette==0.21.0
|
||||
# via fastapi
|
||||
tabulate==0.8.10
|
||||
# via databricks-cli
|
||||
threadpoolctl==3.1.0
|
||||
# via scikit-learn
|
||||
tifffile==2021.11.2
|
||||
@ -380,27 +257,11 @@ typing-extensions==4.3.0
|
||||
urllib3==1.26.10
|
||||
# via
|
||||
# botocore
|
||||
# dulwich
|
||||
# requests
|
||||
# sentry-sdk
|
||||
vega-datasets==0.9.0
|
||||
# via -r requirements.in
|
||||
wandb==0.12.21
|
||||
# via -r requirements.in
|
||||
wcwidth==0.2.5
|
||||
# via prompt-toolkit
|
||||
websocket-client==1.3.3
|
||||
# via
|
||||
# comet-ml
|
||||
# docker
|
||||
werkzeug==2.1.2
|
||||
# via flask
|
||||
wrapt==1.14.1
|
||||
# via comet-ml
|
||||
wurlitzer==3.0.2
|
||||
# via comet-ml
|
||||
zipp==3.8.1
|
||||
# via importlib-metadata
|
||||
|
||||
# The following packages are considered to be unsafe in a requirements file:
|
||||
# setuptools
|
||||
|
||||
@ -15,10 +15,8 @@ from functools import partial
|
||||
from string import capwords
|
||||
from unittest.mock import patch
|
||||
|
||||
import mlflow
|
||||
import pytest
|
||||
import uvicorn
|
||||
import wandb
|
||||
import websockets
|
||||
from fastapi.testclient import TestClient
|
||||
|
||||
@ -196,62 +194,6 @@ class TestBlocksMethods:
|
||||
assert difference >= 0.01
|
||||
assert result
|
||||
|
||||
def test_integration_wandb(self):
|
||||
with captured_output() as (out, err):
|
||||
wandb.log = mock.MagicMock()
|
||||
wandb.Html = mock.MagicMock()
|
||||
demo = gr.Blocks()
|
||||
with demo:
|
||||
gr.Textbox("Hi there!")
|
||||
demo.integrate(wandb=wandb)
|
||||
|
||||
assert (
|
||||
out.getvalue().strip()
|
||||
== "The WandB integration requires you to `launch(share=True)` first."
|
||||
)
|
||||
demo.share_url = "tmp"
|
||||
demo.integrate(wandb=wandb)
|
||||
wandb.log.assert_called_once()
|
||||
|
||||
@mock.patch("comet_ml.Experiment")
|
||||
def test_integration_comet(self, mock_experiment):
|
||||
experiment = mock_experiment()
|
||||
experiment.log_text = mock.MagicMock()
|
||||
experiment.log_other = mock.MagicMock()
|
||||
|
||||
demo = gr.Blocks()
|
||||
with demo:
|
||||
gr.Textbox("Hi there!")
|
||||
|
||||
demo.launch(prevent_thread_lock=True)
|
||||
demo.integrate(comet_ml=experiment)
|
||||
experiment.log_text.assert_called_with("gradio: " + demo.local_url)
|
||||
demo.share_url = "tmp" # used to avoid creating real share links.
|
||||
demo.integrate(comet_ml=experiment)
|
||||
experiment.log_text.assert_called_with("gradio: " + demo.share_url)
|
||||
assert experiment.log_other.call_count == 2
|
||||
demo.share_url = None
|
||||
demo.close()
|
||||
|
||||
def test_integration_mlflow(self):
|
||||
mlflow.log_param = mock.MagicMock()
|
||||
demo = gr.Blocks()
|
||||
with demo:
|
||||
gr.Textbox("Hi there!")
|
||||
|
||||
demo.launch(prevent_thread_lock=True)
|
||||
demo.integrate(mlflow=mlflow)
|
||||
mlflow.log_param.assert_called_with(
|
||||
"Gradio Interface Local Link", demo.local_url
|
||||
)
|
||||
demo.share_url = "tmp" # used to avoid creating real share links.
|
||||
demo.integrate(mlflow=mlflow)
|
||||
mlflow.log_param.assert_called_with(
|
||||
"Gradio Interface Share Link", demo.share_url
|
||||
)
|
||||
demo.share_url = None
|
||||
demo.close()
|
||||
|
||||
@mock.patch("requests.post")
|
||||
def test_initiated_analytics(self, mock_post):
|
||||
with gr.Blocks(analytics_enabled=True):
|
||||
|
||||
@ -5,10 +5,8 @@ from contextlib import contextmanager
|
||||
from functools import partial
|
||||
from string import capwords
|
||||
|
||||
import mlflow
|
||||
import pytest
|
||||
import requests
|
||||
import wandb
|
||||
from fastapi.testclient import TestClient
|
||||
|
||||
import gradio
|
||||
@ -160,63 +158,6 @@ class TestInterface:
|
||||
assert mock_display.call_count == 2
|
||||
interface.close()
|
||||
|
||||
@mock.patch("comet_ml.Experiment")
|
||||
def test_integration_comet(self, mock_experiment):
|
||||
experiment = mock_experiment()
|
||||
experiment.log_text = mock.MagicMock()
|
||||
experiment.log_other = mock.MagicMock()
|
||||
interface = Interface(lambda x: x, "textbox", "label")
|
||||
interface.launch(prevent_thread_lock=True)
|
||||
interface.integrate(comet_ml=experiment)
|
||||
experiment.log_text.assert_called_with("gradio: " + interface.local_url)
|
||||
interface.share_url = "tmp" # used to avoid creating real share links.
|
||||
interface.integrate(comet_ml=experiment)
|
||||
experiment.log_text.assert_called_with("gradio: " + interface.share_url)
|
||||
assert experiment.log_other.call_count == 2
|
||||
interface.share_url = None
|
||||
interface.close()
|
||||
|
||||
def test_integration_mlflow(self):
|
||||
mlflow.log_param = mock.MagicMock()
|
||||
interface = Interface(lambda x: x, "textbox", "label")
|
||||
interface.launch(prevent_thread_lock=True)
|
||||
interface.integrate(mlflow=mlflow)
|
||||
mlflow.log_param.assert_called_with(
|
||||
"Gradio Interface Local Link", interface.local_url
|
||||
)
|
||||
interface.share_url = "tmp" # used to avoid creating real share links.
|
||||
interface.integrate(mlflow=mlflow)
|
||||
mlflow.log_param.assert_called_with(
|
||||
"Gradio Interface Share Link", interface.share_url
|
||||
)
|
||||
interface.share_url = None
|
||||
interface.close()
|
||||
|
||||
def test_integration_wandb(self):
|
||||
with captured_output() as (out, err):
|
||||
wandb.log = mock.MagicMock()
|
||||
wandb.Html = mock.MagicMock()
|
||||
interface = Interface(lambda x: x, "textbox", "label")
|
||||
interface.width = 500
|
||||
interface.height = 500
|
||||
interface.integrate(wandb=wandb)
|
||||
|
||||
assert (
|
||||
out.getvalue().strip()
|
||||
== "The WandB integration requires you to `launch(share=True)` first."
|
||||
)
|
||||
interface.share_url = "tmp"
|
||||
interface.integrate(wandb=wandb)
|
||||
wandb.log.assert_called_once()
|
||||
|
||||
@mock.patch("requests.post")
|
||||
def test_integration_analytics(self, mock_post):
|
||||
mlflow.log_param = mock.MagicMock()
|
||||
interface = Interface(lambda x: x, "textbox", "label")
|
||||
interface.analytics_enabled = True
|
||||
interface.integrate(mlflow=mlflow)
|
||||
mock_post.assert_called_once()
|
||||
|
||||
|
||||
class TestTabbedInterface:
|
||||
def test_tabbed_interface_config_matches_manual_tab(self):
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user