Skip to content

Commit eeb198a

Browse files
author
rainiera
committed
Add 'phrases' to ALLOWED_FIELDS and fix relevant test
1 parent c768c8a commit eeb198a

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

scaleapi/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
'lineannotation': {'attachment', 'attachment_type', 'instruction',
2121
'objects_to_annotate', 'with_labels', 'examples'},
2222
'datacollection': {'attachment', 'attachment_type', 'fields'},
23-
'audiotranscription': {'attachment', 'attachment_type', 'verbatim'}}
23+
'audiotranscription': {'attachment', 'attachment_type', 'verbatim', 'phrases'}}
2424
SCALE_ENDPOINT = 'https://api.scaleapi.com/v1/'
2525
DEFAULT_LIMIT = 100
2626
DEFAULT_OFFSET = 0

tests/test_client.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,8 @@ def test_audiotranscription_ok():
186186
callback_url='http://www.example.com/callback',
187187
attachment_type='audio',
188188
attachment='https://storage.googleapis.com/deepmind-media/pixie/knowing-what-to-say/second-list/speaker-3.wav',
189-
verbatim=False
189+
verbatim=False,
190+
phrases=['avocado', 'stone']
190191
)
191192

192193

0 commit comments

Comments
 (0)