Skip to content

Commit b894bdf

Browse files
fix(pre_commit): 🎨 auto format pre-commit hooks
1 parent adfa656 commit b894bdf

File tree

6 files changed

+73
-42
lines changed

6 files changed

+73
-42
lines changed

tests/cli/test_annotation_handler.py

Lines changed: 15 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -49,13 +49,21 @@ def test_annotation_job_get(self):
4949

5050
def test_annotation_job_create(self):
5151
parser = _build_annotation_parser()
52-
args = parser.parse_args([
53-
"annotation", "job", "create",
54-
"-p", "proj",
55-
"--name", "my-job",
56-
"--batch", "batch-1",
57-
"--assignees", "a@b.com,c@d.com",
58-
])
52+
args = parser.parse_args(
53+
[
54+
"annotation",
55+
"job",
56+
"create",
57+
"-p",
58+
"proj",
59+
"--name",
60+
"my-job",
61+
"--batch",
62+
"batch-1",
63+
"--assignees",
64+
"a@b.com,c@d.com",
65+
]
66+
)
5967
self.assertEqual(args.name, "my-job")
6068
self.assertEqual(args.batch, "batch-1")
6169
self.assertEqual(args.assignees, "a@b.com,c@d.com")

tests/cli/test_deployment_handler.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
"""Tests for the deployment CLI handler."""
22

33
import io
4-
import sys
54
import unittest
65
from unittest.mock import patch
76

@@ -26,18 +25,14 @@ def test_deployment_add_hidden_alias(self) -> None:
2625
from roboflow.cli import build_parser
2726

2827
parser = build_parser()
29-
args = parser.parse_args(
30-
["deployment", "add", "mydepl", "-m", "gpu-small", "-e", "test@example.com"]
31-
)
28+
args = parser.parse_args(["deployment", "add", "mydepl", "-m", "gpu-small", "-e", "test@example.com"])
3229
self.assertIsNotNone(args.func)
3330

3431
def test_deployment_create_canonical(self) -> None:
3532
from roboflow.cli import build_parser
3633

3734
parser = build_parser()
38-
args = parser.parse_args(
39-
["deployment", "create", "mydepl", "-m", "gpu-small", "-e", "test@example.com"]
40-
)
35+
args = parser.parse_args(["deployment", "create", "mydepl", "-m", "gpu-small", "-e", "test@example.com"])
4136
self.assertIsNotNone(args.func)
4237

4338
def test_deployment_machine_type_canonical(self) -> None:

tests/cli/test_image_handler.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -80,9 +80,7 @@ def test_image_delete_parser(self):
8080

8181
def test_image_annotate_parser(self):
8282
parser = _build_image_parser()
83-
args = parser.parse_args(
84-
["image", "annotate", "img-1", "-p", "proj", "--annotation-file", "ann.txt"]
85-
)
83+
args = parser.parse_args(["image", "annotate", "img-1", "-p", "proj", "--annotation-file", "ann.txt"])
8684
self.assertEqual(args.image_id, "img-1")
8785
self.assertEqual(args.annotation_file, "ann.txt")
8886

tests/cli/test_infer_handler.py

Lines changed: 14 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -34,13 +34,20 @@ def test_infer_all_flags(self) -> None:
3434
from roboflow.cli import build_parser
3535

3636
parser = build_parser()
37-
args = parser.parse_args([
38-
"infer", "img.png",
39-
"-m", "proj/1",
40-
"-c", "0.7",
41-
"-o", "0.3",
42-
"-t", "object-detection",
43-
])
37+
args = parser.parse_args(
38+
[
39+
"infer",
40+
"img.png",
41+
"-m",
42+
"proj/1",
43+
"-c",
44+
"0.7",
45+
"-o",
46+
"0.3",
47+
"-t",
48+
"object-detection",
49+
]
50+
)
4451
self.assertAlmostEqual(args.confidence, 0.7)
4552
self.assertAlmostEqual(args.overlap, 0.3)
4653
self.assertEqual(args.type, "object-detection")

tests/cli/test_model_handler.py

Lines changed: 26 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -38,12 +38,18 @@ def test_model_upload_parser(self) -> None:
3838
from roboflow.cli import build_parser
3939

4040
parser = build_parser()
41-
args = parser.parse_args([
42-
"model", "upload",
43-
"-p", "proj1",
44-
"-t", "yolov8",
45-
"-m", "/path/to/model",
46-
])
41+
args = parser.parse_args(
42+
[
43+
"model",
44+
"upload",
45+
"-p",
46+
"proj1",
47+
"-t",
48+
"yolov8",
49+
"-m",
50+
"/path/to/model",
51+
]
52+
)
4753
self.assertEqual(args.project, ["proj1"])
4854
self.assertEqual(args.model_type, "yolov8")
4955
self.assertEqual(args.model_path, "/path/to/model")
@@ -54,12 +60,20 @@ def test_model_upload_multiple_projects(self) -> None:
5460
from roboflow.cli import build_parser
5561

5662
parser = build_parser()
57-
args = parser.parse_args([
58-
"model", "upload",
59-
"-p", "proj1", "-p", "proj2",
60-
"-t", "yolov8",
61-
"-m", "/path/to/model",
62-
])
63+
args = parser.parse_args(
64+
[
65+
"model",
66+
"upload",
67+
"-p",
68+
"proj1",
69+
"-p",
70+
"proj2",
71+
"-t",
72+
"yolov8",
73+
"-m",
74+
"/path/to/model",
75+
]
76+
)
6377
self.assertEqual(args.project, ["proj1", "proj2"])
6478

6579

tests/cli/test_train_handler.py

Lines changed: 15 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -42,12 +42,21 @@ def test_train_optional_args(self) -> None:
4242
from roboflow.cli import build_parser
4343

4444
parser = build_parser()
45-
args = parser.parse_args([
46-
"train", "-p", "proj", "-v", "1",
47-
"--checkpoint", "abc123",
48-
"--speed", "fast",
49-
"--epochs", "50",
50-
])
45+
args = parser.parse_args(
46+
[
47+
"train",
48+
"-p",
49+
"proj",
50+
"-v",
51+
"1",
52+
"--checkpoint",
53+
"abc123",
54+
"--speed",
55+
"fast",
56+
"--epochs",
57+
"50",
58+
]
59+
)
5160
self.assertEqual(args.checkpoint, "abc123")
5261
self.assertEqual(args.speed, "fast")
5362
self.assertEqual(args.epochs, 50)

0 commit comments

Comments
 (0)