Skip to content

Commit

Permalink
Remove unused DFINE model configurations from unit tests
Browse files Browse the repository at this point in the history
  • Loading branch information
eugene123tw committed Jan 6, 2025
1 parent 138ac03 commit 04f1205
Showing 1 changed file with 0 additions and 9 deletions.
9 changes: 0 additions & 9 deletions tests/unit/algo/detection/test_dfine.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,9 +21,6 @@ class TestDFine:
@pytest.mark.parametrize(
"model",
[
DFine(label_info=3, model_name="dfine_hgnetv2_n"),
DFine(label_info=3, model_name="dfine_hgnetv2_s"),
DFine(label_info=3, model_name="dfine_hgnetv2_m"),
DFine(label_info=3, model_name="dfine_hgnetv2_l"),
DFine(label_info=3, model_name="dfine_hgnetv2_x"),
],
Expand All @@ -36,9 +33,6 @@ def test_loss(self, model, fxt_data_module):
@pytest.mark.parametrize(
"model",
[
DFine(label_info=3, model_name="dfine_hgnetv2_n"),
DFine(label_info=3, model_name="dfine_hgnetv2_s"),
DFine(label_info=3, model_name="dfine_hgnetv2_m"),
DFine(label_info=3, model_name="dfine_hgnetv2_l"),
DFine(label_info=3, model_name="dfine_hgnetv2_x"),
],
Expand All @@ -53,9 +47,6 @@ def test_predict(self, model, fxt_data_module):
@pytest.mark.parametrize(
"model",
[
DFine(label_info=3, model_name="dfine_hgnetv2_n"),
DFine(label_info=3, model_name="dfine_hgnetv2_s"),
DFine(label_info=3, model_name="dfine_hgnetv2_m"),
DFine(label_info=3, model_name="dfine_hgnetv2_l"),
DFine(label_info=3, model_name="dfine_hgnetv2_x"),
],
Expand Down

0 comments on commit 04f1205

Please sign in to comment.