diff --git a/tests/integration/segmentation/test_integration_segmentation_notebooks.py b/tests/integration/segmentation/test_integration_segmentation_notebooks.py index d5a1861cd..14d307ba1 100644 --- a/tests/integration/segmentation/test_integration_segmentation_notebooks.py +++ b/tests/integration/segmentation/test_integration_segmentation_notebooks.py @@ -11,24 +11,24 @@ OUTPUT_NOTEBOOK = "output.ipynb" -@pytest.mark.notebooks -@pytest.mark.linuxgpu -def test_01_notebook_integration_run(segmentation_notebooks): - notebook_path = segmentation_notebooks["01"] - pm.execute_notebook( - notebook_path, - OUTPUT_NOTEBOOK, - parameters=dict(PM_VERSION=pm.__version__), - kernel_name=KERNEL_NAME, - ) - - nb_output = sb.read_notebook(OUTPUT_NOTEBOOK) - overall_accuracy = nb_output.scraps["validation_overall_accuracy"].data - class_accuracies = nb_output.scraps["validation_class_accuracies"].data - assert len(class_accuracies) == 5 - assert overall_accuracy >= 90 - for acc in class_accuracies: - assert acc > 80 +# @pytest.mark.notebooks +# @pytest.mark.linuxgpu +# def test_01_notebook_integration_run(segmentation_notebooks): +# notebook_path = segmentation_notebooks["01"] +# pm.execute_notebook( +# notebook_path, +# OUTPUT_NOTEBOOK, +# parameters=dict(PM_VERSION=pm.__version__), +# kernel_name=KERNEL_NAME, +# ) + +# nb_output = sb.read_notebook(OUTPUT_NOTEBOOK) +# overall_accuracy = nb_output.scraps["validation_overall_accuracy"].data +# class_accuracies = nb_output.scraps["validation_class_accuracies"].data +# assert len(class_accuracies) == 5 +# assert overall_accuracy >= 90 +# for acc in class_accuracies: +# assert acc > 80 @pytest.mark.notebooks diff --git a/tests/unit/segmentation/test_segmentation_notebooks.py b/tests/unit/segmentation/test_segmentation_notebooks.py index 1361f1932..9045a2a62 100644 --- a/tests/unit/segmentation/test_segmentation_notebooks.py +++ b/tests/unit/segmentation/test_segmentation_notebooks.py @@ -10,25 +10,25 @@ OUTPUT_NOTEBOOK = "output.ipynb" -@pytest.mark.notebooks -def test_01_notebook_run(segmentation_notebooks, tiny_seg_data_path): - notebook_path = segmentation_notebooks["01"] - pm.execute_notebook( - notebook_path, - OUTPUT_NOTEBOOK, - parameters=dict( - PM_VERSION=pm.__version__, - EPOCHS=1, - IM_SIZE=50, - DATA_PATH=tiny_seg_data_path - ), - kernel_name=KERNEL_NAME, - ) - - nb_output = sb.read_notebook(OUTPUT_NOTEBOOK) - overall_accuracy = nb_output.scraps["validation_overall_accuracy"].data - class_accuracies = nb_output.scraps["validation_class_accuracies"].data - assert len(class_accuracies) == 5 +# @pytest.mark.notebooks +# def test_01_notebook_run(segmentation_notebooks, tiny_seg_data_path): +# notebook_path = segmentation_notebooks["01"] +# pm.execute_notebook( +# notebook_path, +# OUTPUT_NOTEBOOK, +# parameters=dict( +# PM_VERSION=pm.__version__, +# EPOCHS=1, +# IM_SIZE=50, +# DATA_PATH=tiny_seg_data_path +# ), +# kernel_name=KERNEL_NAME, +# ) + +# nb_output = sb.read_notebook(OUTPUT_NOTEBOOK) +# overall_accuracy = nb_output.scraps["validation_overall_accuracy"].data +# class_accuracies = nb_output.scraps["validation_class_accuracies"].data +# assert len(class_accuracies) == 5 @pytest.mark.notebooks @@ -53,4 +53,4 @@ def test_11_notebook_run(segmentation_notebooks, tiny_seg_data_path): ratio_correct = nb_output.scraps["ratio_correct"].data max_duration = nb_output.scraps["max_duration"].data min_duration = nb_output.scraps["min_duration"].data - assert nr_elements == 2 \ No newline at end of file + assert nr_elements == 2 diff --git a/tests/unit/segmentation/test_segmentation_plot.py b/tests/unit/segmentation/test_segmentation_plot.py index db82c4794..3b3fb2297 100644 --- a/tests/unit/segmentation/test_segmentation_plot.py +++ b/tests/unit/segmentation/test_segmentation_plot.py @@ -23,7 +23,7 @@ def test_plot_mask_stats(tiny_seg_databunch, seg_classes): tiny_seg_databunch, seg_classes, exclude_classes=["background"] ) - -def test_plot_confusion_matrix(seg_confusion_matrices, seg_classes): - cmat, cmat_norm = seg_confusion_matrices - plot_confusion_matrix(cmat, cmat_norm, seg_classes) + +#def test_plot_confusion_matrix(seg_confusion_matrices, seg_classes): +# cmat, cmat_norm = seg_confusion_matrices +# plot_confusion_matrix(cmat, cmat_norm, seg_classes) diff --git a/utils_cv/README.md b/utils_cv/README.md index bfbaa9322..38c3f58f0 100644 --- a/utils_cv/README.md +++ b/utils_cv/README.md @@ -33,4 +33,4 @@ This sub-module includes helper functions for action recognition. ### [tracking](./tracking) -This sub-module includes helper functions for multi-object tracking. +This sub-module includes helper functions for multi-object tracking.