Upload dataset_utils.py with huggingface_hub
Browse files- dataset_utils.py +5 -31
dataset_utils.py
CHANGED
@@ -1,12 +1,11 @@
|
|
1 |
-
import os
|
2 |
-
|
3 |
from .artifact import Artifact, UnitxtArtifactNotFoundError, fetch_artifact
|
4 |
from .logging_utils import get_logger
|
|
|
5 |
from .register import _reset_env_local_catalogs, register_all_artifacts
|
|
|
6 |
|
7 |
logger = get_logger()
|
8 |
-
|
9 |
-
__default_recipe__ = "standard_recipe"
|
10 |
|
11 |
|
12 |
def fetch(artifact_name):
|
@@ -18,32 +17,7 @@ def fetch(artifact_name):
|
|
18 |
|
19 |
|
20 |
def parse(query: str):
|
21 |
-
|
22 |
-
result = {}
|
23 |
-
kvs = query.split(",")
|
24 |
-
if len(kvs) == 0:
|
25 |
-
raise ValueError(
|
26 |
-
'Illegal query: "{query}" should contain at least one assignment of the form: key1=value1,key2=value2'
|
27 |
-
)
|
28 |
-
for kv in kvs:
|
29 |
-
key_val = kv.split("=")
|
30 |
-
if (
|
31 |
-
len(key_val) != 2
|
32 |
-
or len(key_val[0].strip()) == 0
|
33 |
-
or len(key_val[1].strip()) == 0
|
34 |
-
):
|
35 |
-
raise ValueError(
|
36 |
-
f'Illegal query: "{query}" with wrong assignment "{kv}" should be of the form: key=value.'
|
37 |
-
)
|
38 |
-
key, val = key_val
|
39 |
-
if val.isdigit():
|
40 |
-
result[key] = int(val)
|
41 |
-
elif val.replace(".", "", 1).isdigit():
|
42 |
-
result[key] = float(val)
|
43 |
-
else:
|
44 |
-
result[key] = val
|
45 |
-
|
46 |
-
return result
|
47 |
|
48 |
|
49 |
def get_dataset_artifact(dataset_str):
|
@@ -53,6 +27,6 @@ def get_dataset_artifact(dataset_str):
|
|
53 |
if recipe is None:
|
54 |
args = parse(dataset_str)
|
55 |
if "type" not in args:
|
56 |
-
args["type"] =
|
57 |
recipe = Artifact.from_dict(args)
|
58 |
return recipe
|
|
|
|
|
|
|
1 |
from .artifact import Artifact, UnitxtArtifactNotFoundError, fetch_artifact
|
2 |
from .logging_utils import get_logger
|
3 |
+
from .parsing_utils import parse_key_equals_value_string_to_dict
|
4 |
from .register import _reset_env_local_catalogs, register_all_artifacts
|
5 |
+
from .settings_utils import get_settings
|
6 |
|
7 |
logger = get_logger()
|
8 |
+
settings = get_settings()
|
|
|
9 |
|
10 |
|
11 |
def fetch(artifact_name):
|
|
|
17 |
|
18 |
|
19 |
def parse(query: str):
|
20 |
+
return parse_key_equals_value_string_to_dict(query)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
21 |
|
22 |
|
23 |
def get_dataset_artifact(dataset_str):
|
|
|
27 |
if recipe is None:
|
28 |
args = parse(dataset_str)
|
29 |
if "type" not in args:
|
30 |
+
args["type"] = settings.default_recipe
|
31 |
recipe = Artifact.from_dict(args)
|
32 |
return recipe
|