deepspeed

1
# Copyright (c) Microsoft Corporation.
2
# SPDX-License-Identifier: Apache-2.0
3

4
# DeepSpeed Team
5

6
import argparse
7
from packaging import version as pkg_version
8

9
parser = argparse.ArgumentParser()
10

11
parser.add_argument("--current_version",
12
                    type=str,
13
                    help="The current version being published to help set the next version.")
14

15
args = parser.parse_args()
16

17
current_version = pkg_version.parse(args.current_version)
18

19
with open('./version.txt', 'w') as fd:
20
    fd.write(f'{current_version.major}.{current_version.minor}.{current_version.micro + 1}\n')
21

22
print(f'{current_version} -> {current_version.major}.{current_version.minor}.{current_version.micro + 1}')
23