mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-11-22 01:12:59 +00:00
index.py: call static function name run
This commit is contained in:
parent
a7ed58fc03
commit
b53c77a2cb
1 changed files with 1 additions and 8 deletions
9
aimodel/src/index.py
Normal file → Executable file
9
aimodel/src/index.py
Normal file → Executable file
|
@ -16,14 +16,7 @@ def main():
|
||||||
return
|
return
|
||||||
|
|
||||||
imported_module = importlib.import_module(f"subcommands.{subcommand}")
|
imported_module = importlib.import_module(f"subcommands.{subcommand}")
|
||||||
# TODO: Support multiple subcommands here
|
imported_module.run(args)
|
||||||
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__":
|
if __name__ == "__main__":
|
||||||
|
|
Loading…
Reference in a new issue