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
|
||||
|
||||
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)
|
||||
|
||||
imported_module.run(args)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
|
Loading…
Reference in a new issue