Skip to content
Snippets Groups Projects
Commit 42765c91 authored by Manon Blanco's avatar Manon Blanco
Browse files

Remove useless `validate_uuid` function

parent 5f99f175
No related branches found
No related tags found
1 merge request!323Remove useless `validate_uuid` function
...@@ -5,6 +5,7 @@ Extract dataset from Arkindex using a corpus export. ...@@ -5,6 +5,7 @@ Extract dataset from Arkindex using a corpus export.
import argparse import argparse
import pathlib import pathlib
from typing import Union
from uuid import UUID from uuid import UUID
from dan.datasets.extract.arkindex import run from dan.datasets.extract.arkindex import run
...@@ -12,18 +13,15 @@ from dan.datasets.extract.arkindex import run ...@@ -12,18 +13,15 @@ from dan.datasets.extract.arkindex import run
MANUAL_SOURCE = "manual" MANUAL_SOURCE = "manual"
def validate_uuid(arg_uuid): def parse_worker_version(worker_version_id) -> Union[str, bool]:
try:
return UUID(arg_uuid)
except ValueError:
raise argparse.ArgumentTypeError(f"`{arg_uuid}` is not a valid UUID.")
def parse_worker_version(worker_version_id):
if worker_version_id == MANUAL_SOURCE: if worker_version_id == MANUAL_SOURCE:
return False return False
validate_uuid(worker_version_id) try:
UUID(worker_version_id)
except ValueError:
raise argparse.ArgumentTypeError(f"`{worker_version_id}` is not a valid UUID.")
return worker_version_id return worker_version_id
...@@ -79,19 +77,19 @@ def add_extract_parser(subcommands) -> None: ...@@ -79,19 +77,19 @@ def add_extract_parser(subcommands) -> None:
parser.add_argument( parser.add_argument(
"--train-folder", "--train-folder",
type=validate_uuid, type=UUID,
help="ID of the training folder to extract from Arkindex.", help="ID of the training folder to extract from Arkindex.",
required=True, required=True,
) )
parser.add_argument( parser.add_argument(
"--val-folder", "--val-folder",
type=validate_uuid, type=UUID,
help="ID of the validation folder to extract from Arkindex.", help="ID of the validation folder to extract from Arkindex.",
required=True, required=True,
) )
parser.add_argument( parser.add_argument(
"--test-folder", "--test-folder",
type=validate_uuid, type=UUID,
help="ID of the testing folder to extract from Arkindex.", help="ID of the testing folder to extract from Arkindex.",
required=True, required=True,
) )
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment