Upload dataset.py with huggingface_hub
Browse files- dataset.py +19 -8
dataset.py
CHANGED
|
@@ -11,6 +11,7 @@ from .common import __file__ as _
|
|
| 11 |
from .dataclass import __file__ as _
|
| 12 |
from .dict_utils import __file__ as _
|
| 13 |
from .file_utils import __file__ as _
|
|
|
|
| 14 |
from .fusion import __file__ as _
|
| 15 |
from .generator_utils import __file__ as _
|
| 16 |
from .hf_utils import __file__ as _
|
|
@@ -27,9 +28,11 @@ from .random_utils import __file__ as _
|
|
| 27 |
from .recipe import __file__ as _
|
| 28 |
from .register import __file__ as _
|
| 29 |
from .register import register_all_artifacts
|
|
|
|
| 30 |
from .schema import __file__ as _
|
| 31 |
from .split_utils import __file__ as _
|
| 32 |
from .splitters import __file__ as _
|
|
|
|
| 33 |
from .stream import __file__ as _
|
| 34 |
from .task import __file__ as _
|
| 35 |
from .templates import __file__ as _
|
|
@@ -54,14 +57,22 @@ def parse(query: str):
|
|
| 54 |
Parses a query of the form 'key1=value1,key2=value2,...' into a dictionary.
|
| 55 |
"""
|
| 56 |
result = {}
|
| 57 |
-
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
|
| 61 |
-
|
| 62 |
-
|
| 63 |
-
|
| 64 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 65 |
|
| 66 |
return result
|
| 67 |
|
|
|
|
| 11 |
from .dataclass import __file__ as _
|
| 12 |
from .dict_utils import __file__ as _
|
| 13 |
from .file_utils import __file__ as _
|
| 14 |
+
from .formats import __file__ as _
|
| 15 |
from .fusion import __file__ as _
|
| 16 |
from .generator_utils import __file__ as _
|
| 17 |
from .hf_utils import __file__ as _
|
|
|
|
| 28 |
from .recipe import __file__ as _
|
| 29 |
from .register import __file__ as _
|
| 30 |
from .register import register_all_artifacts
|
| 31 |
+
from .renderers import __file__ as _
|
| 32 |
from .schema import __file__ as _
|
| 33 |
from .split_utils import __file__ as _
|
| 34 |
from .splitters import __file__ as _
|
| 35 |
+
from .standard import __file__ as _
|
| 36 |
from .stream import __file__ as _
|
| 37 |
from .task import __file__ as _
|
| 38 |
from .templates import __file__ as _
|
|
|
|
| 57 |
Parses a query of the form 'key1=value1,key2=value2,...' into a dictionary.
|
| 58 |
"""
|
| 59 |
result = {}
|
| 60 |
+
kvs = query.split(",")
|
| 61 |
+
if len(kvs) == 0:
|
| 62 |
+
raise ValueError(
|
| 63 |
+
'Illegal query: "{query}" should contain at least one assignment of the form: key1=value1,key2=value2'
|
| 64 |
+
)
|
| 65 |
+
for kv in kvs:
|
| 66 |
+
key_val = kv.split("=")
|
| 67 |
+
if len(key_val) != 2 or len(key_val[0].strip()) == 0 or len(key_val[1].strip()) == 0:
|
| 68 |
+
raise ValueError('Illegal query: "{query}" with wrong assignment "{kv}" should be of the form: key=value.')
|
| 69 |
+
key, val = key_val
|
| 70 |
+
if val.isdigit():
|
| 71 |
+
result[key] = int(val)
|
| 72 |
+
elif val.replace(".", "", 1).isdigit():
|
| 73 |
+
result[key] = float(val)
|
| 74 |
+
else:
|
| 75 |
+
result[key] = val
|
| 76 |
|
| 77 |
return result
|
| 78 |
|