fix missing imports, add console scripts, fix quality checks

Author: markurtz

setup.py

@@ -118,7 +118,12 @@ def _setup_extras() -> Dict:
 
 
 def _setup_entry_points() -> Dict:
-    return {}
+    return {
+        "console_scripts": [
+            "sparseml.framework=sparseml.framework.info:_main",
+            "sparseml.sparsification=sparseml.sparsification.info:_main",
+        ]
+    }
 
 
 def _setup_long_description() -> Tuple[str, str]:

src/sparseml/__init__.py

@@ -30,5 +30,16 @@
     detect_framework,
     execute_in_sparseml_framework,
 )
-from .framework import FrameworkInferenceProviderInfo, FrameworkInfo, framework_info
-from .sparsification import SparsificationInfo, sparsification_info
+from .framework import (
+    FrameworkInferenceProviderInfo,
+    FrameworkInfo,
+    framework_info,
+    save_framework_info,
+    load_framework_info,
+)
+from .sparsification import (
+    SparsificationInfo,
+    sparsification_info,
+    save_sparsification_info,
+    load_sparsification_info,
+)

tests/sparseml/test_imports.py

@@ -14,11 +14,7 @@
 
 
 def test_imports():
-    # logging imports
-    # sparsification imports
-    # framework imports
-    # base imports
-    # version imports
+    # flake8: noqa
     from sparseml import (
         Framework,
         FrameworkInferenceProviderInfo,