mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-07-04 20:24:55 +00:00
34 lines
674 B
Python
34 lines
674 B
Python
#!/usr/bin/env python3
|
|
|
|
import importlib
|
|
import sys
|
|
|
|
from loguru import logger
|
|
|
|
def init_logging():
|
|
pass
|
|
|
|
from parse_args import parse_args
|
|
|
|
def main():
|
|
subcommand, args = parse_args()
|
|
if args == None:
|
|
return
|
|
|
|
imported_module = importlib.import_module(f"subcommands.{subcommand}")
|
|
# TODO: Support multiple subcommands here
|
|
match subcommand:
|
|
case "pretrain":
|
|
imported_module.pretrain(args)
|
|
case _:
|
|
sys.stderr.write(f"Error: The subcommand '{subcommand}' hasn't been registered in index.py yet.\nThis is a bug.\n")
|
|
exit(1)
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|
|
else:
|
|
print("This script must be run directly. It cannot be imported.")
|
|
exit(1)
|