SYMBOL INDEX (477 symbols across 53 files) FILE: noxfile.py function tests (line 17) | def tests(session: nox.Session) -> None: function core_tests (line 29) | def core_tests(session: nox.Session) -> None: function lint (line 42) | def lint(session: nox.Session) -> None: function safety (line 52) | def safety(session: nox.Session) -> None: function build (line 60) | def build(session: nox.Session) -> None: function dev (line 68) | def dev(session: nox.Session) -> None: FILE: src/dvclive/dvc.py function _dvc_dir (line 20) | def _dvc_dir(dirname: StrPath) -> str: function _find_dvc_root (line 24) | def _find_dvc_root(root: Optional[StrPath] = None) -> Optional[str]: function get_dvc_repo (line 43) | def get_dvc_repo() -> Optional["Repo"]: function make_dvcyaml (line 58) | def make_dvcyaml(live) -> None: # noqa: C901 function update_dvcyaml (line 93) | def update_dvcyaml(live, updates): function get_exp_name (line 133) | def get_exp_name(name, scm, baseline_rev) -> str: function find_overlapping_stage (line 161) | def find_overlapping_stage(dvc_repo: "Repo", path: StrPath) -> Optional[... function ensure_dir_is_tracked (line 170) | def ensure_dir_is_tracked(directory: str, dvc_repo: "Repo") -> None: FILE: src/dvclive/error.py class DvcLiveError (line 4) | class DvcLiveError(Exception): class InvalidDataTypeError (line 8) | class InvalidDataTypeError(DvcLiveError): method __init__ (line 9) | def __init__(self, name, val): class InvalidDvcyamlError (line 15) | class InvalidDvcyamlError(DvcLiveError): method __init__ (line 16) | def __init__(self): class InvalidImageNameError (line 20) | class InvalidImageNameError(DvcLiveError): method __init__ (line 21) | def __init__(self, name): class InvalidPlotTypeError (line 26) | class InvalidPlotTypeError(DvcLiveError): method __init__ (line 27) | def __init__(self, name): class InvalidParameterTypeError (line 37) | class InvalidParameterTypeError(DvcLiveError): method __init__ (line 38) | def __init__(self, msg: Any): class InvalidReportModeError (line 42) | class InvalidReportModeError(DvcLiveError): method __init__ (line 43) | def __init__(self, val): FILE: src/dvclive/fabric.py class DVCLiveLogger (line 35) | class DVCLiveLogger(Logger): method __init__ (line 38) | def __init__( method name (line 52) | def name(self) -> str: method version (line 56) | def version(self) -> Union[int, str]: method experiment (line 63) | def experiment(self) -> "Live": method log_metrics (line 78) | def log_metrics( method log_hyperparams (line 112) | def log_hyperparams(self, params: Union[dict[str, Any], Namespace]) ->... method finalize (line 124) | def finalize(self, status: str) -> None: # noqa: ARG002 method _sanitize_params (line 129) | def _sanitize_params(params: Union[dict[str, Any], Namespace]) -> dict... method __getstate__ (line 145) | def __getstate__(self) -> dict[str, Any]: FILE: src/dvclive/fastai.py function _inside_fine_tune (line 10) | def _inside_fine_tune(): class DVCLiveCallback (line 27) | class DVCLiveCallback(Callback): method __init__ (line 28) | def __init__( method before_fit (line 39) | def before_fit(self): method after_epoch (line 52) | def after_epoch(self): method after_fit (line 69) | def after_fit(self): FILE: src/dvclive/huggingface.py class DVCLiveCallback (line 20) | class DVCLiveCallback(TrainerCallback): method __init__ (line 21) | def __init__( method on_train_begin (line 36) | def on_train_begin( method on_log (line 45) | def on_log( method on_save (line 57) | def on_save( method on_train_end (line 68) | def on_train_end( FILE: src/dvclive/keras.py class DVCLiveCallback (line 10) | class DVCLiveCallback(tf.keras.callbacks.Callback): method __init__ (line 11) | def __init__( method on_epoch_end (line 21) | def on_epoch_end(self, epoch: int, logs: Optional[dict] = None): method on_train_end (line 27) | def on_train_end(self, logs: Optional[dict] = None): FILE: src/dvclive/lgbm.py class DVCLiveCallback (line 6) | class DVCLiveCallback: method __init__ (line 7) | def __init__(self, live: Optional[Live] = None, **kwargs): method __call__ (line 11) | def __call__(self, env): FILE: src/dvclive/lightning.py function _should_sync (line 32) | def _should_sync(): class DVCLiveLogger (line 50) | class DVCLiveLogger(Logger, FabricDVCLiveLogger): method __init__ (line 51) | def __init__( method log_metrics (line 71) | def log_metrics( method after_save_checkpoint (line 81) | def after_save_checkpoint(self, checkpoint_callback: ModelCheckpoint) ... method finalize (line 92) | def finalize(self, status: str) -> None: method _scan_checkpoints (line 103) | def _scan_checkpoints(self, checkpoint_callback: ModelCheckpoint) -> N... method _save_checkpoints (line 112) | def _save_checkpoints(self, checkpoint_callback: ModelCheckpoint) -> N... FILE: src/dvclive/live.py class Live (line 78) | class Live: method __init__ (line 79) | def __init__( method _init_resume (line 183) | def _init_resume(self): method _init_cleanup (line 192) | def _init_cleanup(self): method _init_dvc (line 211) | def _init_dvc(self): # noqa: C901 method _init_dvc_file (line 267) | def _init_dvc_file(self) -> str: method _init_dvc_pipeline (line 282) | def _init_dvc_pipeline(self): method _init_studio (line 296) | def _init_studio(self): method _init_report (line 310) | def _init_report(self): method _init_test (line 335) | def _init_test(self): method dir (line 355) | def dir(self) -> str: method params_file (line 360) | def params_file(self) -> str: method metrics_file (line 364) | def metrics_file(self) -> str: method dvc_file (line 368) | def dvc_file(self) -> str: method plots_dir (line 373) | def plots_dir(self) -> str: method artifacts_dir (line 377) | def artifacts_dir(self) -> str: method report_file (line 381) | def report_file(self) -> Optional[str]: method step (line 388) | def step(self) -> int: method step (line 392) | def step(self, value: int) -> None: method monitor_system (line 396) | def monitor_system( method sync (line 433) | def sync(self): method next_step (line 443) | def next_step(self): method log_metric (line 459) | def log_metric( method log_image (line 507) | def log_image( method log_plot (line 569) | def log_plot( method log_sklearn_plot (line 629) | def log_sklearn_plot( method _read_params (line 694) | def _read_params(self): method _dump_params (line 699) | def _dump_params(self): method log_params (line 705) | def log_params(self, params: dict[str, ParamLike]): method log_param (line 724) | def log_param(self, name: str, val: ParamLike): method log_artifact (line 741) | def log_artifact( method cache (line 828) | def cache(self, path): method make_summary (line 859) | def make_summary(self): method make_report (line 876) | def make_report(self): method make_dvcyaml (line 893) | def make_dvcyaml(self): method _get_live_data (line 904) | def _get_live_data(self) -> Optional[dict[str, Any]]: method post_data_to_studio (line 926) | def post_data_to_studio(self): method _wait_for_studio_updates_posted (line 947) | def _wait_for_studio_updates_posted(self): method end (line 952) | def end(self): method read_step (line 1000) | def read_step(self): method read_latest (line 1004) | def read_latest(self): method __enter__ (line 1010) | def __enter__(self): method __exit__ (line 1014) | def __exit__(self, exc_type, exc_val, exc_tb): method save_dvc_exp (line 1019) | def save_dvc_exp(self): FILE: src/dvclive/monitor_system.py class _SystemMonitor (line 49) | class _SystemMonitor: method __init__ (line 58) | def __init__( method _check_interval (line 82) | def _check_interval(self, interval: float, max_interval: float) -> float: method _check_num_samples (line 91) | def _check_num_samples( method _check_directories_to_monitor (line 104) | def _check_directories_to_monitor( method _monitoring_loop (line 117) | def _monitoring_loop(self): method _get_metrics (line 148) | def _get_metrics(self) -> dict[str, Union[float, int]]: method _get_ram_info (line 156) | def _get_ram_info(self) -> dict[str, Union[float, int]]: method _get_cpu_info (line 164) | def _get_cpu_info(self) -> dict[str, Union[float, int]]: method _get_disk_info (line 181) | def _get_disk_info(self) -> dict[str, Union[float, int]]: method _get_gpu_info (line 204) | def _get_gpu_info(self) -> dict[str, Union[float, int]]: method end (line 240) | def end(self): FILE: src/dvclive/optuna.py class DVCLiveCallback (line 5) | class DVCLiveCallback: method __init__ (line 6) | def __init__(self, metric_name="metric", **kwargs) -> None: method __call__ (line 12) | def __call__(self, study, trial) -> None: method _log_metrics (line 17) | def _log_metrics(self, values, live): FILE: src/dvclive/plots/base.py class Data (line 5) | class Data(abc.ABC): method __init__ (line 6) | def __init__(self, name: str, output_folder: str) -> None: method step (line 12) | def step(self) -> int: method step (line 16) | def step(self, val: int) -> None: method output_path (line 21) | def output_path(self) -> Path: method subfolder (line 26) | def subfolder(self): method could_log (line 31) | def could_log(val) -> bool: method dump (line 35) | def dump(self, val, **kwargs): FILE: src/dvclive/plots/custom.py class CustomPlot (line 9) | class CustomPlot(Data): method __init__ (line 13) | def __init__( method output_path (line 40) | def output_path(self) -> Path: method could_log (line 46) | def could_log(val: object) -> bool: method plot_config (line 50) | def plot_config(self): method dump (line 53) | def dump(self, val, **kwargs) -> None: # noqa: ARG002 FILE: src/dvclive/plots/image.py class Image (line 8) | class Image(Data): method output_path (line 13) | def output_path(self) -> Path: method could_log (line 19) | def could_log(val: object) -> bool: method dump (line 30) | def dump(self, val, **kwargs) -> None: # noqa: ARG002 FILE: src/dvclive/plots/metric.py class Metric (line 10) | class Metric(Data): method could_log (line 15) | def could_log(val: object) -> bool: method output_path (line 24) | def output_path(self) -> Path: method dump (line 29) | def dump(self, val, **kwargs) -> None: method summary_keys (line 46) | def summary_keys(self) -> list[str]: FILE: src/dvclive/plots/sklearn.py class SKLearnPlot (line 6) | class SKLearnPlot(CustomPlot): method could_log (line 10) | def could_log(val: object) -> bool: class Roc (line 14) | class Roc(SKLearnPlot): method __init__ (line 15) | def __init__(self, name: str, output_folder: str, **plot_config) -> None: method dump (line 26) | def dump(self, val, **kwargs) -> None: class PrecisionRecall (line 41) | class PrecisionRecall(SKLearnPlot): method __init__ (line 42) | def __init__(self, name: str, output_folder: str, **plot_config) -> None: method dump (line 51) | def dump(self, val, **kwargs) -> None: class Det (line 67) | class Det(SKLearnPlot): method __init__ (line 68) | def __init__(self, name: str, output_folder: str, **plot_config) -> None: method dump (line 79) | def dump(self, val, **kwargs) -> None: class ConfusionMatrix (line 95) | class ConfusionMatrix(SKLearnPlot): method __init__ (line 96) | def __init__(self, name: str, output_folder: str, **plot_config) -> None: method dump (line 109) | def dump(self, val, **kwargs) -> None: # noqa: ARG002 class Calibration (line 117) | class Calibration(SKLearnPlot): method __init__ (line 118) | def __init__(self, name: str, output_folder: str, **plot_config) -> None: method dump (line 129) | def dump(self, val, **kwargs) -> None: FILE: src/dvclive/plots/utils.py class NumpyEncoder (line 19) | class NumpyEncoder(json.JSONEncoder): method default (line 20) | def default(self, o): FILE: src/dvclive/report.py function get_scalar_renderers (line 30) | def get_scalar_renderers(metrics_path): function get_image_renderers (line 48) | def get_image_renderers(images_folder): function get_custom_plot_renderers (line 66) | def get_custom_plot_renderers(plots_folder, live): function get_sklearn_plot_renderers (line 84) | def get_sklearn_plot_renderers(plots_folder, live): function get_metrics_renderers (line 110) | def get_metrics_renderers(dvclive_summary): function get_params_renderers (line 122) | def get_params_renderers(dvclive_params): function make_report (line 134) | def make_report(live: "Live"): FILE: src/dvclive/serialize.py class YAMLError (line 8) | class YAMLError(DvcLiveError): class YAMLFileCorruptedError (line 12) | class YAMLFileCorruptedError(YAMLError): method __init__ (line 13) | def __init__(self, path): function load_yaml (line 17) | def load_yaml(path, typ="safe"): function get_yaml (line 29) | def get_yaml(): function dump_yaml (line 41) | def dump_yaml(content, output_file): function dump_json (line 48) | def dump_json(content, output_file, indent=4, **kwargs): function make_dir (line 55) | def make_dir(output_file): FILE: src/dvclive/studio.py function _cast_to_numbers (line 25) | def _cast_to_numbers(datapoints: Mapping): function _adapt_path (line 41) | def _adapt_path(live: "Live", name: StrPath): function _adapt_image (line 49) | def _adapt_image(image_path: StrPath): function _adapt_images (line 54) | def _adapt_images(live: "Live", images: "list[Image]"): function _get_studio_updates (line 62) | def _get_studio_updates(live: "Live", data: dict[str, Any]): function get_dvc_studio_config (line 90) | def get_dvc_studio_config(live: "Live"): function increment_num_points_sent_to_studio (line 97) | def increment_num_points_sent_to_studio(live, plots_sent, data): function post_to_studio (line 108) | def post_to_studio( # noqa: C901 FILE: src/dvclive/utils.py function run_once (line 31) | def run_once(f): function open_file_in_browser (line 43) | def open_file_in_browser(file) -> bool: function env2bool (line 50) | def env2bool(var, undefined=False): function standardize_metric_name (line 60) | def standardize_metric_name(metric_name: str, framework: str) -> str: function parse_tsv (line 96) | def parse_tsv(path): function parse_json (line 102) | def parse_json(path): function parse_metrics (line 107) | def parse_metrics(live): function matplotlib_installed (line 119) | def matplotlib_installed() -> bool: function inside_colab (line 127) | def inside_colab() -> bool: function inside_notebook (line 135) | def inside_notebook() -> bool: function clean_and_copy_into (line 151) | def clean_and_copy_into(src: StrPath, dst: StrPath) -> str: function isinstance_without_import (line 170) | def isinstance_without_import(val, module, name): function catch_and_warn (line 177) | def catch_and_warn(exception, logger, on_finally=None): function rel_path (line 193) | def rel_path(path, dvc_root_path): function read_history (line 198) | def read_history(live, metric): function read_latest (line 211) | def read_latest(live, metric_name): function convert_datapoints_to_list_of_dicts (line 216) | def convert_datapoints_to_list_of_dicts( FILE: src/dvclive/vscode.py function _dvc_exps_run_dir (line 11) | def _dvc_exps_run_dir(dirname: StrPath) -> str: function _dvclive_only_signal_file (line 15) | def _dvclive_only_signal_file(root_dir: StrPath) -> str: function _dvclive_step_completed_signal_file (line 20) | def _dvclive_step_completed_signal_file(root_dir: StrPath) -> str: function _find_non_queue_root (line 25) | def _find_non_queue_root() -> Optional[str]: function _write_file (line 29) | def _write_file(file: str, contents: dict[str, Union[str, int]]): function mark_dvclive_step_completed (line 40) | def mark_dvclive_step_completed(step: int) -> None: function cleanup_dvclive_step_completed (line 59) | def cleanup_dvclive_step_completed() -> None: function mark_dvclive_only_started (line 73) | def mark_dvclive_only_started(exp_name: str) -> None: function mark_dvclive_only_ended (line 90) | def mark_dvclive_only_ended() -> None: FILE: src/dvclive/xgb.py class DVCLiveCallback (line 10) | class DVCLiveCallback(TrainingCallback): method __init__ (line 11) | def __init__( method after_iteration (line 27) | def after_iteration(self, model, epoch, evals_log): method after_training (line 35) | def after_training(self, model): FILE: tests/conftest.py function tmp_dir (line 10) | def tmp_dir(tmp_path, monkeypatch): function mocked_dvc_repo (line 16) | def mocked_dvc_repo(tmp_dir, mocker): function mocked_dvc_subrepo (line 31) | def mocked_dvc_subrepo(tmp_dir, mocker, mocked_dvc_repo): function dvc_repo (line 37) | def dvc_repo(tmp_dir): function _capture_wrap (line 48) | def _capture_wrap(): function _mocked_webbrowser_open (line 55) | def _mocked_webbrowser_open(mocker): function _mocked_ci (line 60) | def _mocked_ci(monkeypatch): function mocked_studio_post (line 65) | def mocked_studio_post(mocker, monkeypatch): FILE: tests/frameworks/test_fabric.py class BoringModel (line 15) | class BoringModel(torch.nn.Module): method __init__ (line 16) | def __init__(self): method forward (line 20) | def forward(self, x): function test_dvclive_log_metrics (line 26) | def test_dvclive_log_metrics(tmp_path, mocked_dvc_repo, step_idx): function test_dvclive_log_hyperparams (line 37) | def test_dvclive_log_hyperparams(tmp_path, mocked_dvc_repo): function test_dvclive_finalize (line 54) | def test_dvclive_finalize(monkeypatch, tmp_path, mocked_dvc_repo): FILE: tests/frameworks/test_fastai.py function data_loader (line 25) | def data_loader(): function test_fastai_callback (line 44) | def test_fastai_callback(tmp_dir, data_loader, mocker): function test_fastai_pass_logger (line 71) | def test_fastai_pass_logger(): function test_fast_ai_resume (line 78) | def test_fast_ai_resume(tmp_dir, data_loader, mocker): function test_fast_ai_avoid_unnecessary_end_calls (line 97) | def test_fast_ai_avoid_unnecessary_end_calls(tmp_dir, data_loader, mocker): function test_fastai_save_model_callback (line 114) | def test_fastai_save_model_callback(tmp_dir, data_loader, mocker): FILE: tests/frameworks/test_huggingface.py function compute_metrics (line 27) | def compute_metrics(eval_preds): class RegressionDataset (line 38) | class RegressionDataset: method __init__ (line 39) | def __init__(self, a=2, b=3, length=64, seed=42, label_names=None): method __len__ (line 50) | def __len__(self): method __getitem__ (line 53) | def __getitem__(self, i): class RegressionModelConfig (line 59) | class RegressionModelConfig(PretrainedConfig): method __init__ (line 60) | def __init__(self, a=0, b=0, double_output=False, random_torch=True, *... class RegressionPreTrainedModel (line 69) | class RegressionPreTrainedModel(PreTrainedModel): method __init__ (line 73) | def __init__(self, config): method forward (line 79) | def forward(self, input_x, labels=None, **kwargs): function data (line 88) | def data(): function model (line 93) | def model(): function args (line 99) | def args(): function test_huggingface_integration (line 111) | def test_huggingface_integration(tmp_dir, model, args, data, mocker, cal... function test_huggingface_log_model (line 144) | def test_huggingface_log_model( function test_huggingface_pass_logger (line 196) | def test_huggingface_pass_logger(callback): function test_huggingface_report_to (line 204) | def test_huggingface_report_to(model, report_to): FILE: tests/frameworks/test_keras.py function xor_model (line 16) | def xor_model(): function test_keras_callback (line 37) | def test_keras_callback(tmp_dir, xor_model, mocker): function test_keras_callback_pass_logger (line 61) | def test_keras_callback_pass_logger(): FILE: tests/frameworks/test_lgbm.py function model_params (line 21) | def model_params(): function iris_data (line 26) | def iris_data(): function test_lgbm_integration (line 37) | def test_lgbm_integration(tmp_dir, model_params, iris_data): function test_lgbm_integration_multi_eval (line 59) | def test_lgbm_integration_multi_eval(tmp_dir, model_params, iris_data): function test_lgbm_pass_logger (line 84) | def test_lgbm_pass_logger(): FILE: tests/frameworks/test_lightning.py class XORDataset (line 31) | class XORDataset(Dataset): method __init__ (line 32) | def __init__(self, *args, **kwargs): method __getitem__ (line 36) | def __getitem__(self, index): method __len__ (line 41) | def __len__(self): class LitXOR (line 45) | class LitXOR(LightningModule): method __init__ (line 46) | def __init__( method forward (line 60) | def forward(self, *args, **kwargs): method train_loader (line 69) | def train_loader(self): method train_dataloader (line 73) | def train_dataloader(self): method training_step (line 76) | def training_step(self, *args, **kwargs): method configure_optimizers (line 91) | def configure_optimizers(self): method predict_dataloader (line 94) | def predict_dataloader(self): method test_dataloader (line 97) | def test_dataloader(self): method val_dataloader (line 100) | def val_dataloader(self): function test_lightning_integration (line 104) | def test_lightning_integration(tmp_dir, mocker): function test_lightning_default_dir (line 143) | def test_lightning_default_dir(tmp_dir): function test_lightning_kwargs (line 158) | def test_lightning_kwargs(tmp_dir): function test_lightning_log_model (line 180) | def test_lightning_log_model(tmp_dir, mocker, log_model, save_top_k): function test_lightning_steps (line 213) | def test_lightning_steps(tmp_dir, mocker): class ValLitXOR (line 245) | class ValLitXOR(LitXOR): method val_loader (line 246) | def val_loader(self): method val_dataloader (line 250) | def val_dataloader(self): method training_step (line 253) | def training_step(self, *args, **kwargs): method validation_step (line 261) | def validation_step(self, *args, **kwargs): function test_lightning_force_init (line 270) | def test_lightning_force_init(tmp_dir, mocker): class TestModel (line 281) | class TestModel(BoringModel): method __init__ (line 282) | def __init__(self, foo, bar=5): function _test_logger_init_args (line 288) | def _test_logger_init_args(logger_name, init, unresolved={}): # noqa: B006 function test_dvclive_logger_init_args (line 313) | def test_dvclive_logger_init_args(): FILE: tests/frameworks/test_optuna.py function objective (line 14) | def objective(trial): function test_optuna_ (line 19) | def test_optuna_(tmp_dir, mocked_dvc_repo): FILE: tests/frameworks/test_xgboost.py function train_params (line 22) | def train_params(): function iris_data (line 27) | def iris_data(): function iris_train_eval_data (line 35) | def iris_train_eval_data(): function test_xgb_integration (line 54) | def test_xgb_integration( function test_xgb_pass_logger (line 82) | def test_xgb_pass_logger(): FILE: tests/plots/test_custom.py function test_log_custom_plot (line 7) | def test_log_custom_plot(tmp_dir): function test_log_custom_plot_multi_y (line 34) | def test_log_custom_plot_multi_y(tmp_dir): function test_log_custom_plot_with_template_as_empty_string (line 61) | def test_log_custom_plot_with_template_as_empty_string(tmp_dir): FILE: tests/plots/test_image.py class ExtendedImage (line 12) | class ExtendedImage(Image.Image): method __init__ (line 13) | def __init__(self, img): method __getattr__ (line 16) | def __getattr__(self, key): function test_pil (line 20) | def test_pil(tmp_dir): function test_pil_omitting_extension_doesnt_save_without_valid_format (line 28) | def test_pil_omitting_extension_doesnt_save_without_valid_format(tmp_dir): function test_pil_omitting_extension_sets_the_format_if_path_given (line 37) | def test_pil_omitting_extension_sets_the_format_if_path_given(tmp_dir): function test_invalid_extension (line 53) | def test_invalid_extension(tmp_dir): function test_numpy (line 63) | def test_numpy(tmp_dir, shape): function test_path (line 77) | def test_path(tmp_dir): function test_override_on_step (line 98) | def test_override_on_step(tmp_dir): function test_cleanup (line 113) | def test_cleanup(tmp_dir): function test_custom_class (line 125) | def test_custom_class(tmp_dir): function test_matplotlib (line 134) | def test_matplotlib(tmp_dir): function test_cache_images (line 149) | def test_cache_images(tmp_dir, dvc_repo, cache): FILE: tests/plots/test_metric.py function test_numpy (line 13) | def test_numpy(tmp_dir, dtype): function test_name_with_dot (line 27) | def test_name_with_dot(tmp_dir): FILE: tests/plots/test_sklearn.py function y_true_y_pred_y_score (line 12) | def y_true_y_pred_y_score(): function test_log_calibration_curve (line 28) | def test_log_calibration_curve(tmp_dir, y_true_y_pred_y_score, mocker): function test_log_det_curve (line 43) | def test_log_det_curve(tmp_dir, y_true_y_pred_y_score, mocker): function test_log_roc_curve (line 57) | def test_log_roc_curve(tmp_dir, y_true_y_pred_y_score, mocker): function test_log_prc_curve (line 71) | def test_log_prc_curve(tmp_dir, y_true_y_pred_y_score, mocker): function test_log_confusion_matrix (line 85) | def test_log_confusion_matrix(tmp_dir, y_true_y_pred_y_score, mocker): function test_dump_kwargs (line 101) | def test_dump_kwargs(tmp_dir, y_true_y_pred_y_score, mocker): function test_override_on_step (line 113) | def test_override_on_step(tmp_dir): function test_cleanup (line 129) | def test_cleanup(tmp_dir, y_true_y_pred_y_score): function test_custom_name (line 144) | def test_custom_name(tmp_dir, y_true_y_pred_y_score): function test_custom_title (line 160) | def test_custom_title(tmp_dir, y_true_y_pred_y_score): function test_custom_labels (line 193) | def test_custom_labels(tmp_dir, y_true_y_pred_y_score): FILE: tests/test_cleanup.py function test_cleanup (line 15) | def test_cleanup(tmp_dir, html, dvcyaml): FILE: tests/test_context_manager.py function test_context_manager (line 7) | def test_context_manager(tmp_dir): function test_context_manager_skips_end_calls (line 21) | def test_context_manager_skips_end_calls(tmp_dir): FILE: tests/test_dvc.py function test_get_dvc_repo (line 14) | def test_get_dvc_repo(tmp_dir): function test_get_dvc_repo_subdir (line 20) | def test_get_dvc_repo_subdir(tmp_dir): function test_exp_save_on_end (line 29) | def test_exp_save_on_end(tmp_dir, save, mocked_dvc_repo): function test_exp_save_skip_on_env_vars (line 45) | def test_exp_save_skip_on_env_vars(tmp_dir, monkeypatch): function test_exp_save_with_dvc_files (line 60) | def test_exp_save_with_dvc_files(tmp_dir, mocker): function test_exp_save_dvcexception_is_ignored (line 83) | def test_exp_save_dvcexception_is_ignored(tmp_dir, mocker): function test_untracked_dvclive_files_inside_dvc_exp_run_are_added (line 98) | def test_untracked_dvclive_files_inside_dvc_exp_run_are_added( function test_dvc_outs_are_not_added (line 116) | def test_dvc_outs_are_not_added(tmp_dir, mocked_dvc_repo, monkeypatch): function test_errors_on_git_add_are_catched (line 135) | def test_errors_on_git_add_are_catched(tmp_dir, mocked_dvc_repo, monkeyp... function test_exp_save_message (line 145) | def test_exp_save_message(tmp_dir, mocked_dvc_repo): function test_exp_save_name (line 156) | def test_exp_save_name(tmp_dir, mocked_dvc_repo): function test_no_scm_repo (line 167) | def test_no_scm_repo(tmp_dir, mocker): function test_dvc_repro (line 179) | def test_dvc_repro(tmp_dir, monkeypatch, mocked_dvc_repo, mocked_studio_... function test_get_exp_name_valid (line 188) | def test_get_exp_name_valid(tmp_dir, mocked_dvc_repo): function test_get_exp_name_random (line 193) | def test_get_exp_name_random(tmp_dir, mocked_dvc_repo, mocker): function test_get_exp_name_invalid (line 201) | def test_get_exp_name_invalid(tmp_dir, mocked_dvc_repo, mocker, caplog): function test_get_exp_name_duplicate (line 211) | def test_get_exp_name_duplicate(tmp_dir, mocked_dvc_repo, mocker, caplog): function test_test_mode (line 223) | def test_test_mode(tmp_dir, monkeypatch, mocked_dvc_repo): FILE: tests/test_log_artifact.py function test_log_artifact (line 21) | def test_log_artifact(tmp_dir, dvc_repo, cache): function test_log_artifact_on_existing_dvc_file (line 30) | def test_log_artifact_on_existing_dvc_file(tmp_dir, dvc_repo): function test_log_artifact_twice (line 45) | def test_log_artifact_twice(tmp_dir, dvc_repo): function test_log_artifact_with_save_dvc_exp (line 54) | def test_log_artifact_with_save_dvc_exp(tmp_dir, mocker, mocked_dvc_repo): function test_log_artifact_type_model (line 68) | def test_log_artifact_type_model(tmp_dir, mocked_dvc_repo): function test_log_artifact_dvc_symlink (line 79) | def test_log_artifact_dvc_symlink(tmp_dir, dvc_repo): function test_log_artifact_copy (line 91) | def test_log_artifact_copy(tmp_dir, dvc_repo): function test_log_artifact_copy_overwrite (line 106) | def test_log_artifact_copy_overwrite(tmp_dir, dvc_repo): function test_log_artifact_copy_directory_overwrite (line 126) | def test_log_artifact_copy_directory_overwrite(tmp_dir, dvc_repo): function test_log_artifact_type_model_provided_name (line 156) | def test_log_artifact_type_model_provided_name(tmp_dir, mocked_dvc_repo): function test_log_artifact_type_model_on_step_and_final (line 167) | def test_log_artifact_type_model_on_step_and_final(tmp_dir, mocked_dvc_r... function test_log_artifact_type_model_on_step (line 183) | def test_log_artifact_type_model_on_step(tmp_dir, mocked_dvc_repo): function test_log_artifact_attrs (line 199) | def test_log_artifact_attrs(tmp_dir, mocked_dvc_repo): function test_log_artifact_type_model_when_dvc_add_fails (line 219) | def test_log_artifact_type_model_when_dvc_add_fails(tmp_dir, mocker, moc... function test_log_artifact_inside_pipeline (line 231) | def test_log_artifact_inside_pipeline(tmp_dir, mocker, dvc_repo, tracked): function test_log_artifact_inside_pipeline_subdir (line 270) | def test_log_artifact_inside_pipeline_subdir(tmp_dir, mocker, dvc_repo): function test_log_artifact_no_repo (line 291) | def test_log_artifact_no_repo(tmp_dir, mocker): function test_log_artifact_invalid_path_type (line 302) | def test_log_artifact_invalid_path_type(invalid_path, tmp_dir): FILE: tests/test_log_metric.py function test_logging_no_step (line 14) | def test_logging_no_step(tmp_dir): function test_logging_step (line 29) | def test_logging_step(tmp_dir, path): function test_nested_logging (line 42) | def test_nested_logging(tmp_dir): function test_log_metric_timestamp (line 70) | def test_log_metric_timestamp(tmp_dir, timestamp): function test_invalid_metric_type (line 81) | def test_invalid_metric_type(tmp_dir, invalid_type): function test_log_metric_inf_nan (line 95) | def test_log_metric_inf_nan(tmp_dir, val): function test_log_metic_str (line 101) | def test_log_metic_str(tmp_dir): FILE: tests/test_log_param.py function test_cleanup_params (line 10) | def test_cleanup_params(tmp_dir): function test_log_param (line 45) | def test_log_param(tmp_dir, param_name, param_value): function test_log_params (line 54) | def test_log_params(tmp_dir): function test_log_params_resume (line 71) | def test_log_params_resume(tmp_dir, resume): function test_log_param_custom_obj (line 79) | def test_log_param_custom_obj(tmp_dir): FILE: tests/test_logging.py function test_logger (line 6) | def test_logger(tmp_dir, mocker): function test_suppress_dvc_logs (line 21) | def test_suppress_dvc_logs(tmp_dir, mocked_dvc_repo): FILE: tests/test_make_dvcyaml.py function test_make_dvcyaml_empty (line 13) | def test_make_dvcyaml_empty(tmp_dir): function test_make_dvcyaml_param (line 20) | def test_make_dvcyaml_param(tmp_dir): function test_make_dvcyaml_metrics (line 30) | def test_make_dvcyaml_metrics(tmp_dir): function test_make_dvcyaml_metrics_no_plots (line 41) | def test_make_dvcyaml_metrics_no_plots(tmp_dir): function test_make_dvcyaml_summary (line 51) | def test_make_dvcyaml_summary(tmp_dir): function test_make_dvcyaml_all_plots (line 61) | def test_make_dvcyaml_all_plots(tmp_dir): function test_make_dvcyaml_relpath (line 117) | def test_make_dvcyaml_relpath(tmp_dir, mocked_dvc_repo): function test_make_dvcyaml_update (line 200) | def test_make_dvcyaml_update(tmp_dir, orig_yaml, updated_yaml): function test_make_dvcyaml_update_artifact (line 259) | def test_make_dvcyaml_update_artifact( function test_make_dvcyaml_update_all (line 272) | def test_make_dvcyaml_update_all(tmp_dir, mocked_dvc_repo): function test_make_dvcyaml_update_multiple (line 347) | def test_make_dvcyaml_update_multiple(tmp_dir, mocked_dvc_repo): function test_dvcyaml_on_next_step (line 372) | def test_dvcyaml_on_next_step(tmp_dir, dvcyaml, mocked_dvc_repo): function test_dvcyaml_on_end (line 382) | def test_dvcyaml_on_end(tmp_dir, dvcyaml, mocked_dvc_repo): function test_make_dvcyaml_idempotent (line 391) | def test_make_dvcyaml_idempotent(tmp_dir, mocked_dvc_repo): function test_warn_on_dvcyaml_output_overlap (line 407) | def test_warn_on_dvcyaml_output_overlap(tmp_dir, mocker, mocked_dvc_repo... function test_make_dvcyaml (line 429) | def test_make_dvcyaml(tmp_dir, mocked_dvc_repo, dvcyaml): function test_make_dvcyaml_no_repo (line 443) | def test_make_dvcyaml_no_repo(tmp_dir, mocker): function test_make_dvcyaml_invalid (line 450) | def test_make_dvcyaml_invalid(tmp_dir, mocker): function test_make_dvcyaml_on_end (line 455) | def test_make_dvcyaml_on_end(tmp_dir, mocker): function test_make_dvcyaml_false (line 462) | def test_make_dvcyaml_false(tmp_dir, mocker): function test_make_dvcyaml_none (line 469) | def test_make_dvcyaml_none(tmp_dir, mocker): FILE: tests/test_make_report.py function test_get_image_renderers (line 22) | def test_get_image_renderers(tmp_dir, mode): function test_get_renderers (line 36) | def test_get_renderers(tmp_dir, mocker): function test_report_init (line 71) | def test_report_init(monkeypatch, mocker): function test_make_report (line 92) | def test_make_report(tmp_dir, mode): function test_make_report_open (line 107) | def test_make_report_open(tmp_dir, mocker, monkeypatch): function test_get_plot_renderers_sklearn (line 131) | def test_get_plot_renderers_sklearn(tmp_dir): function test_get_plot_renderers_custom (line 173) | def test_get_plot_renderers_custom(tmp_dir): function test_report_notebook (line 204) | def test_report_notebook(tmp_dir, mocker): FILE: tests/test_make_summary.py function test_make_summary_without_calling_log (line 7) | def test_make_summary_without_calling_log(tmp_dir): function test_make_summary_is_called_on_end (line 21) | def test_make_summary_is_called_on_end(tmp_dir): function test_make_summary_on_end_dont_increment_step (line 35) | def test_make_summary_on_end_dont_increment_step(tmp_dir): FILE: tests/test_monitor_system.py function mock_psutil_cpu (line 30) | def mock_psutil_cpu(mocker): function mock_psutil_ram (line 38) | def mock_psutil_ram(mocker): function mock_psutil_disk (line 48) | def mock_psutil_disk(mocker): function mock_psutil_disk_with_oserror (line 56) | def mock_psutil_disk_with_oserror(mocker): function mock_pynvml (line 72) | def mock_pynvml(mocker, num_gpus=2): function cpu_metrics (line 93) | def cpu_metrics(): function _timeserie_schema (line 111) | def _timeserie_schema(name, value): function cpu_timeseries (line 116) | def cpu_timeseries(): function gpu_timeseries (line 136) | def gpu_timeseries(): function test_monitor_system_is_false (line 147) | def test_monitor_system_is_false(tmp_dir, mocker): function test_monitor_system_is_true (line 159) | def test_monitor_system_is_true(tmp_dir, mocker): function test_all_threads_close (line 172) | def test_all_threads_close(tmp_dir, mocker): function test_ignore_non_existent_directories (line 195) | def test_ignore_non_existent_directories(tmp_dir, mocker): function test_monitor_system_metrics (line 223) | def test_monitor_system_metrics(tmp_dir, cpu_metrics, mocker): function test_monitor_system_timeseries (line 247) | def test_monitor_system_timeseries(tmp_dir, cpu_timeseries, mocker): function test_monitor_system_metrics_with_gpu (line 274) | def test_monitor_system_metrics_with_gpu(tmp_dir, cpu_metrics, mocker): function test_monitor_system_timeseries_with_gpu (line 307) | def test_monitor_system_timeseries_with_gpu( FILE: tests/test_post_to_studio.py function get_studio_call (line 19) | def get_studio_call(event_type, exp_name, **kwargs): function test_post_to_studio (line 38) | def test_post_to_studio(tmp_dir, mocked_dvc_repo, mocked_studio_post): function test_post_to_studio_subrepo (line 95) | def test_post_to_studio_subrepo(tmp_dir, mocked_dvc_subrepo, mocked_stud... function test_post_to_studio_repo_url (line 107) | def test_post_to_studio_repo_url(tmp_dir, dvc_repo, mocked_studio_post, ... function test_post_to_studio_failed_data_request (line 118) | def test_post_to_studio_failed_data_request( function test_post_to_studio_failed_start_request (line 157) | def test_post_to_studio_failed_start_request( function test_post_to_studio_done_only_once (line 176) | def test_post_to_studio_done_only_once(tmp_dir, mocked_dvc_repo, mocked_... function test_post_to_studio_snapshots_data_to_send (line 196) | def test_post_to_studio_snapshots_data_to_send( function test_studio_updates_posted_on_end (line 251) | def test_studio_updates_posted_on_end(tmp_path, mocked_dvc_repo, mocked_... function test_studio_update_raises_exception (line 275) | def test_studio_update_raises_exception(tmp_path, mocked_dvc_repo, mocke... function test_post_to_studio_skip_start_and_done_on_env_var (line 306) | def test_post_to_studio_skip_start_and_done_on_env_var( function test_post_to_studio_dvc_studio_config (line 325) | def test_post_to_studio_dvc_studio_config( function test_post_to_studio_skip_if_no_token (line 348) | def test_post_to_studio_skip_if_no_token( function test_post_to_studio_shorten_names (line 371) | def test_post_to_studio_shorten_names(tmp_dir, mocked_dvc_repo, mocked_s... function test_post_to_studio_inside_dvc_exp (line 395) | def test_post_to_studio_inside_dvc_exp( function test_post_to_studio_inside_subdir (line 417) | def test_post_to_studio_inside_subdir( function test_post_to_studio_inside_subdir_dvc_exp (line 446) | def test_post_to_studio_inside_subdir_dvc_exp( function test_post_to_studio_without_exp (line 477) | def test_post_to_studio_without_exp(tmp_dir, mocked_dvc_repo, mocked_stu... function test_get_dvc_studio_config_none (line 481) | def test_get_dvc_studio_config_none(mocker): function test_get_dvc_studio_config_env_var (line 487) | def test_get_dvc_studio_config_env_var(monkeypatch, mocker): function test_get_dvc_studio_config_dvc_repo (line 499) | def test_get_dvc_studio_config_dvc_repo(mocked_dvc_repo): function test_post_to_studio_images (line 509) | def test_post_to_studio_images(tmp_dir, mocked_dvc_repo, mocked_studio_p... function test_post_to_studio_message (line 533) | def test_post_to_studio_message(tmp_dir, mocked_dvc_repo, mocked_studio_... function test_post_to_studio_name (line 544) | def test_post_to_studio_name(tmp_dir, mocked_dvc_repo, mocked_studio_post): function test_post_to_studio_if_done_skipped (line 555) | def test_post_to_studio_if_done_skipped(tmp_dir, mocked_dvc_repo, mocked... function test_post_to_studio_no_repo (line 571) | def test_post_to_studio_no_repo(tmp_dir, monkeypatch, mocked_studio_post): function test_post_to_studio_skip_if_no_repo_url (line 629) | def test_post_to_studio_skip_if_no_repo_url( function test_post_to_studio_repeat_step (line 648) | def test_post_to_studio_repeat_step(tmp_dir, mocked_dvc_repo, mocked_stu... FILE: tests/test_resume.py function test_resume (line 12) | def test_resume(tmp_dir, resume, steps, metrics): function test_resume_on_first_init (line 39) | def test_resume_on_first_init(tmp_dir): function test_resume_env_var (line 45) | def test_resume_env_var(tmp_dir, monkeypatch): FILE: tests/test_step.py function test_allow_step_override (line 10) | def test_allow_step_override(tmp_dir, metric): function test_custom_steps (line 17) | def test_custom_steps(tmp_dir): function test_log_reset_with_set_step (line 32) | def test_log_reset_with_set_step(tmp_dir): function test_get_step_resume (line 51) | def test_get_step_resume(tmp_dir): function test_get_step_custom_steps (line 67) | def test_get_step_custom_steps(tmp_dir): function test_get_step_control_flow (line 79) | def test_get_step_control_flow(tmp_dir): function test_set_step_only (line 91) | def test_set_step_only(tmp_dir): function test_step_on_end (line 100) | def test_step_on_end(tmp_dir): FILE: tests/test_utils.py function test_standardize_metric_name (line 19) | def test_standardize_metric_name(framework, logged, standardized): function test_convert_datapoints_to_list_of_dicts (line 39) | def test_convert_datapoints_to_list_of_dicts(input_data, expected_output): function test_unsupported_format (line 43) | def test_unsupported_format(): FILE: tests/test_vscode.py function test_vscode_dvclive_step_completed_signal_file (line 11) | def test_vscode_dvclive_step_completed_signal_file( function test_vscode_dvclive_only_signal_file (line 63) | def test_vscode_dvclive_only_signal_file(tmp_dir, dvc_root, mocker):