avanigupta commited on
Commit
081eb22
·
1 Parent(s): c3002ad

expand datasets

Browse files
Files changed (1) hide show
  1. dataqa_env/server/tasks.py +38 -3
dataqa_env/server/tasks.py CHANGED
@@ -95,7 +95,17 @@ def create_task_easy(seed: int = 42) -> Task:
95
  107,Grace Lee,grace.lee@company.com,Marketing,75000,2021-12-01
96
  108,Hank Brown,hank.brown@company.com,Sales,65000,2023-04-18
97
  109,Iris Patel,iris.patel@company.com,HR,73000,2020-02-28
98
- 110,Jack Taylor,jack.taylor@company.com,Engineering,97000,2022-09-14"""
 
 
 
 
 
 
 
 
 
 
99
 
100
  schema_desc = """Columns:
101
  - employee_id: integer, unique, range 100-999
@@ -186,7 +196,17 @@ ORD-016,CUST-114,Bluetooth Speaker,Electronics,1,49.99,2024-01-30,UK,delivered,4
186
  ORD-017,CUST-115,Jump Rope,Sports,2,8.99,2024-01-31,US,shipped,17.98
187
  ORD-018,CUST-116,Coffee Table Book,Books,1,32.00,2024-02-01,CA,delivered,32.00
188
  ORD-019,CUST-117,Ergonomic Chair,Home,1,450.00,2024-02-02,US,processing,450.00
189
- ORD-020,CUST-118,Fitness Tracker,Electronics,1,79.99,2024-02-03,AU,delivered,79.99"""
 
 
 
 
 
 
 
 
 
 
190
 
191
  schema_desc = """Columns:
192
  - order_id: string, unique, format ORD-NNN
@@ -290,7 +310,22 @@ EXP-011,yolov5-m,coco-2017,118287,5000,40670,0.01,32,300,0.032,0.045,0.0,10.2,24
290
  EXP-012,wav2vec2,librispeech,281241,5567,2620,0.0001,8,20,0.92,1.05,0.0,12.6,15.0,2024-03-13T11:30:00
291
  EXP-013,clip-base,cc3m,2818102,15000,15000,0.00001,256,32,2.15,2.38,0.0,22.4,48.0,2024-03-14T08:00:00
292
  EXP-014,detr,coco-2017,118287,5000,40670,0.0001,4,500,1.85,2.12,0.0,16.0,72.0,2024-03-15T10:00:00
293
- EXP-015,whisper-small,common-voice,520000,16000,16000,0.00005,16,5,0.55,0.68,0.0,7.4,6.5,2024-03-16T14:00:00"""
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
294
 
295
  schema_desc = """Columns:
296
  - experiment_id: string, unique, format EXP-NNN
 
95
  107,Grace Lee,grace.lee@company.com,Marketing,75000,2021-12-01
96
  108,Hank Brown,hank.brown@company.com,Sales,65000,2023-04-18
97
  109,Iris Patel,iris.patel@company.com,HR,73000,2020-02-28
98
+ 110,Jack Taylor,jack.taylor@company.com,Engineering,97000,2022-09-14
99
+ 111,Kevin Zhang,kevin.zhang@company.com,Engineering,91000,2021-05-22
100
+ 112,Laura Adams,laura.adams@company.com,Sales,69000,2022-11-03
101
+ 113,Mike Torres,mike.torres@company.com,Marketing,74000,2020-08-17
102
+ 114,Nina Sharma,nina.sharma@company.com,HR,76000,2019-04-30
103
+ 115,Oscar Rivera,oscar.rivera@company.com,Engineering,105000,2018-12-10
104
+ 116,Paula Green,paula.green@company.com,Sales,67000,2023-06-25
105
+ 117,Quinn Murphy,quinn.murphy@company.com,Marketing,78000,2021-03-08
106
+ 118,Rosa Diaz,rosa.diaz@company.com,Engineering,99000,2022-01-19
107
+ 119,Sam Cooper,sam.cooper@company.com,HR,70000,2020-10-05
108
+ 120,Tara Singh,tara.singh@company.com,Sales,66000,2023-02-14"""
109
 
110
  schema_desc = """Columns:
111
  - employee_id: integer, unique, range 100-999
 
196
  ORD-017,CUST-115,Jump Rope,Sports,2,8.99,2024-01-31,US,shipped,17.98
197
  ORD-018,CUST-116,Coffee Table Book,Books,1,32.00,2024-02-01,CA,delivered,32.00
198
  ORD-019,CUST-117,Ergonomic Chair,Home,1,450.00,2024-02-02,US,processing,450.00
199
+ ORD-020,CUST-118,Fitness Tracker,Electronics,1,79.99,2024-02-03,AU,delivered,79.99
200
+ ORD-021,CUST-119,Laptop Sleeve,Electronics,1,24.99,2024-02-04,US,delivered,24.99
201
+ ORD-022,CUST-120,Hiking Backpack,Sports,1,65.00,2024-02-05,CA,shipped,65.00
202
+ ORD-023,CUST-121,Machine Learning Book,Books,1,54.99,2024-02-06,UK,delivered,54.99
203
+ ORD-024,CUST-122,Plant Pot Set,Home,3,15.00,2024-02-07,US,delivered,45.00
204
+ ORD-025,CUST-123,Noise Cancelling Headphones,Electronics,1,199.99,2024-02-08,DE,shipped,199.99
205
+ ORD-026,CUST-124,Basketball,Sports,1,29.99,2024-02-09,US,delivered,29.99
206
+ ORD-027,CUST-125,Cookbook Collection,Books,2,22.50,2024-02-10,AU,delivered,45.00
207
+ ORD-028,CUST-126,Smart Plug,Home,4,12.99,2024-02-11,US,processing,51.96
208
+ ORD-029,CUST-127,Wireless Charger,Electronics,1,34.99,2024-02-12,UK,delivered,34.99
209
+ ORD-030,CUST-128,Dumbbells Set,Sports,1,89.00,2024-02-13,US,shipped,89.00"""
210
 
211
  schema_desc = """Columns:
212
  - order_id: string, unique, format ORD-NNN
 
310
  EXP-012,wav2vec2,librispeech,281241,5567,2620,0.0001,8,20,0.92,1.05,0.0,12.6,15.0,2024-03-13T11:30:00
311
  EXP-013,clip-base,cc3m,2818102,15000,15000,0.00001,256,32,2.15,2.38,0.0,22.4,48.0,2024-03-14T08:00:00
312
  EXP-014,detr,coco-2017,118287,5000,40670,0.0001,4,500,1.85,2.12,0.0,16.0,72.0,2024-03-15T10:00:00
313
+ EXP-015,whisper-small,common-voice,520000,16000,16000,0.00005,16,5,0.55,0.68,0.0,7.4,6.5,2024-03-16T14:00:00
314
+ EXP-016,mobilenet-v3,imagenet-1k,1281167,50000,100000,0.004,128,150,0.92,1.05,72.8,4.1,18.0,2024-03-17T08:30:00
315
+ EXP-017,albert-base,mnli,392702,9815,9796,0.00002,32,5,0.32,0.41,83.1,6.2,1.8,2024-03-18T11:00:00
316
+ EXP-018,gpt-neo-1.3b,pile-subset,1500000,50000,50000,0.0002,8,2,2.85,2.98,0.0,18.5,36.0,2024-03-19T14:00:00
317
+ EXP-019,swin-tiny,imagenet-1k,1281167,50000,100000,0.001,256,300,0.78,0.95,78.2,8.6,42.0,2024-03-20T09:00:00
318
+ EXP-020,deberta-large,squad-v2,130319,11873,8862,0.00001,16,5,0.35,0.42,85.7,15.2,4.5,2024-03-21T10:30:00
319
+ EXP-021,yolov8-s,coco-2017,118287,5000,40670,0.01,64,200,0.028,0.038,0.0,6.8,16.0,2024-03-22T13:00:00
320
+ EXP-022,bart-base,xsum,204045,11332,11334,0.0001,32,10,1.22,1.38,0.0,8.4,6.2,2024-03-23T15:30:00
321
+ EXP-023,convnext-tiny,imagenet-1k,1281167,50000,100000,0.002,256,300,0.74,0.92,79.5,7.2,38.0,2024-03-24T08:00:00
322
+ EXP-024,xlm-roberta,xnli,392702,2490,5010,0.00002,16,10,0.41,0.48,82.3,12.4,5.8,2024-03-25T11:00:00
323
+ EXP-025,stable-diffusion,laion-400m,400000000,10000,10000,0.0001,4,1,0.45,0.52,0.0,24.0,168.0,2024-03-26T09:00:00
324
+ EXP-026,phi-2,dolly-15k,15011,500,500,0.00005,8,3,0.82,0.95,0.0,10.2,2.5,2024-03-27T14:00:00
325
+ EXP-027,dino-v2,imagenet-1k,1281167,50000,100000,0.0005,64,100,0.42,0.58,0.0,11.8,28.0,2024-03-28T10:00:00
326
+ EXP-028,electra-small,glue-mrpc,3668,408,1725,0.0001,32,10,0.38,0.44,87.2,3.8,0.8,2024-03-29T16:00:00
327
+ EXP-029,sam-base,sa-1b,11000000,50000,50000,0.0001,4,1,0.95,1.08,0.0,16.4,96.0,2024-03-30T08:00:00
328
+ EXP-030,llama2-13b,oasst1,84437,4401,4401,0.00001,2,3,0.78,0.88,0.0,52.0,12.0,2024-03-31T12:00:00"""
329
 
330
  schema_desc = """Columns:
331
  - experiment_id: string, unique, format EXP-NNN