Skip to content

Commit 4082008

Browse files
authored
Remove torchao_nightly package (#661)
* Remove torchao_nightly package * test * test
1 parent 8c6b4f9 commit 4082008

File tree

2 files changed

+6
-7
lines changed

2 files changed

+6
-7
lines changed

.github/workflows/build_wheels_linux.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ on:
66
paths:
77
- build/packaging/**
88
- .github/workflows/build_wheels_linux.yml
9+
- setup.py
910
push:
1011
branches:
1112
- nightly

setup.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
from setuptools import find_packages, setup
1212

13-
current_date = datetime.now().strftime("%Y.%m.%d")
13+
current_date = datetime.now().strftime("%Y%m%d")
1414

1515

1616
def get_git_commit_id():
@@ -27,8 +27,6 @@ def read_version(file_path="version.txt"):
2727
with open(file_path, "r") as file:
2828
return file.readline().strip()
2929

30-
# Determine the package name based on the presence of an environment variable
31-
package_name = "torchao-nightly" if os.environ.get("TORCHAO_NIGHTLY") else "torchao"
3230

3331
# Use Git commit ID if VERSION_SUFFIX is not set
3432
version_suffix = os.getenv("VERSION_SUFFIX")
@@ -37,9 +35,9 @@ def read_version(file_path="version.txt"):
3735

3836
use_cpp = os.getenv('USE_CPP')
3937

40-
41-
# Version is year.month.date if using nightlies
42-
version = current_date if package_name == "torchao-nightly" else read_version()
38+
version_prefix = read_version()
39+
# Version is version.dev year month date if using nightlies and version if not
40+
version = f"{version_prefix}.dev{current_date}" if os.environ.get("TORCHAO_NIGHTLY") else version_prefix
4341

4442
import torch
4543

@@ -124,7 +122,7 @@ def get_extensions():
124122
return ext_modules
125123

126124
setup(
127-
name=package_name,
125+
name="torchao",
128126
version=version+version_suffix,
129127
packages=find_packages(),
130128
include_package_data=True,

0 commit comments

Comments
 (0)