Skip to content

Commit

Permalink
argparse
Browse files Browse the repository at this point in the history
  • Loading branch information
alinelena committed Apr 30, 2024
1 parent 7817984 commit 74f9050
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion mace/cli/eval_configs.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,12 +55,14 @@ def parse_args() -> argparse.Namespace:
)
return parser.parse_args()


def main() -> None:
args = parse_args()
run(args)


def run(args: argparse.Namespace) -> None:

torch_tools.set_default_dtype(args.default_dtype)
device = torch_tools.init_device(args.device)

Expand Down

0 comments on commit 74f9050

Please sign in to comment.