File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 33import json
44import logging
55import os
6+ import sys
67from datetime import datetime , timezone
78from pathlib import Path
89from typing import Any , Dict , Optional
2223from aind_data_schema_models .organizations import Organization
2324from pydantic import ValidationError
2425
25- from aind_metadata_mapper .models import JobSettings
2626from aind_metadata_mapper .base import MapperJobSettings
2727from aind_metadata_mapper .mapper_registry import registry
28+ from aind_metadata_mapper .models import JobSettings
2829
2930
3031def _metadata_service_helper (url : str ) -> Optional [dict ]:
@@ -606,8 +607,10 @@ def run_job(self) -> None:
606607
607608
608609if __name__ == "__main__" :
609- # JobSettings has cli_parse_args=True, so it automatically parses command line arguments
610- # and handles converting CLI args to the appropriate types
611- main_job_settings = JobSettings ()
610+ # Allows a user to input job-settings as json string in command line
611+ if len (sys .argv [1 :]) == 2 and sys .argv [1 ] == "--job-settings" :
612+ main_job_settings = JobSettings .model_validate_json (sys .argv [2 ])
613+ else :
614+ main_job_settings = JobSettings ()
612615 job = GatherMetadataJob (settings = main_job_settings )
613616 job .run_job ()
You can’t perform that action at this time.
0 commit comments