Skip to content

Commit f46993a

Browse files
committed
Fix epytext and skip test that is not suitable for classification-only exports
1 parent 2d2ec41 commit f46993a

File tree

2 files changed

+18
-10
lines changed

2 files changed

+18
-10
lines changed

luxonis_ml/data/exporters/fiftyone_classification_exporter.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,9 @@
1414

1515

1616
class FiftyOneClassificationExporter(BaseExporter):
17-
"""Output structure:
18-
{dataset_name}/
17+
"""Output structure::
18+
19+
<dataset_name>/
1920
train/
2021
data/
2122
000001.jpg
@@ -31,15 +32,16 @@ class FiftyOneClassificationExporter(BaseExporter):
3132
...
3233
labels.json
3334
34-
The labels.json has structure:
35-
{
35+
The labels.json has structure::
36+
37+
E{lb}
3638
"classes": ["class1", "class2", ...],
37-
"labels": {
39+
"labels": E{lb}
3840
"000001": 0, # index into classes array
3941
"000002": 1,
4042
...
41-
}
42-
}
43+
E{rb}
44+
E{rb}
4345
"""
4446

4547
def __init__(

tests/test_data/test_export.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,13 +8,19 @@
88
from luxonis_ml.enums.enums import DatasetType
99

1010
# SOLO does not have an associated exporter,
11-
# SEGMASK and CLSDIR return empty directories
11+
# SEGMASK, CLSDIR, and FIFTYONECLS return empty directories
1212
# because COCO_people_subset does not contain
13-
# image-level masks or classes
13+
# image-level masks or classification labels
1414
EXPORT_DATASET_TYPES = [
1515
dt
1616
for dt in list(DatasetType)
17-
if dt not in {DatasetType.SEGMASK, DatasetType.CLSDIR, DatasetType.SOLO}
17+
if dt
18+
not in {
19+
DatasetType.SEGMASK,
20+
DatasetType.CLSDIR,
21+
DatasetType.SOLO,
22+
DatasetType.FIFTYONECLS,
23+
}
1824
]
1925

2026

0 commit comments

Comments
 (0)