initial commit

.github/workflows/build-nightly.yaml

@@ -0,0 +1 @@
+# TODO

makefile → Makefile

@@ -1,4 +1,6 @@
+.PHONY: build docs test
 
+BUILD_ARGS :=  # set nightly to build nightly release
 PYCHECKDIRS := src tests
 PYCHECKGLOBS := 'src/**/*.py' 'tests/**/*.py' 'utils/**/*.py' 'examples/**/*.py' setup.py
 # run checks on all files for the repo
@@ -25,7 +27,8 @@ test:
 
 # creates wheel file
 build:
-	python3 setup.py sdist bdist_wheel $(BUILD_ARGS)
+	@echo "Building the wheel for the repository";
+	python3 setup.py sdist bdist_wheel $(BUILD_ARGS);
 
 # clean package
 clean:

setup.py

@@ -12,10 +12,27 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-
+import os
 from setuptools import setup, find_packages
 from typing import List, Dict, Tuple
 
+# default variables to be overwritten by the version.py file
+is_release = None
+version = "unknown"
+version_major_minor = version
+
+# load and overwrite version and release info from compressed-tensors package
+exec(open(os.path.join("src", "compressed_tensors", "version.py")).read())
+print(f"loaded version {version} from src/compressed_tensors/version.py")
+version_nm_deps = f"{version_major_minor}.0"
+
+if is_release:
+    _PACKAGE_NAME = "compressed-tensors"
+else:
+    _PACKAGE_NAME = "compressed-tensors-nightly"
+
+
+
 def _setup_long_description() -> Tuple[str, str]:
     return open("README.md", "r", encoding="utf-8").read(), "text/markdown"
 
@@ -31,8 +48,8 @@ def _setup_extras() -> Dict:
     return {"dev": ["black==22.12.0", "isort==5.8.0", "wheel>=0.36.2", "flake8>=3.8.3", "pytest>=6.0.0", "nbconvert>=7.16.3"]}
 
 setup(
-    name="compressed-tensors",
-    version="0.3.3",
+    name=_PACKAGE_NAME,
+    version=version,
     author="Neuralmagic, Inc.",
     author_email="support@neuralmagic.com",
     license="Apache 2.0",

src/compressed_tensors/version.py

@@ -0,0 +1,50 @@
+# Copyright (c) 2021 - present / Neuralmagic, Inc. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""
+Functionality for storing and setting the version info for SparseML
+"""
+
+from datetime import date
+
+
+version_base = "0.3.3"
+is_release = False  # change to True to set the generated version as a release version
+
+
+def _generate_version():
+    if is_release:
+        return version_base
+    else:
+        return f"{version_base}.{date.today().strftime('%Y%m%d')}"
+
+
+__all__ = [
+    "__version__",
+    "version_base",
+    "is_release",
+    "version",
+    "version_major",
+    "version_minor",
+    "version_patch",
+    "version_build",
+    "version_major_minor",
+]
+__version__ = _generate_version()
+
+version = __version__
+version_major, version_minor, version_patch, version_build = version.split(".") + (
+    [None] if len(version.split(".")) < 4 else []
+)  # handle conditional for version being 3 parts or 4 (4 containing build date)
+version_major_minor = f"{version_major}.{version_minor}"