Merge branch 'main' into interface-framework-onnx

Author: markurtz

Makefile

@@ -6,6 +6,7 @@ CHECKGLOBS := 'examples/**/*.py' 'integrations/**/*.py' 'scripts/**/*.py' 'src/*
 DOCDIR := docs
 MDCHECKGLOBS := 'docs/**/*.md' 'docs/**/*.rst' 'examples/**/*.md' 'integrations/**/*.md' 'notebooks/**/*.md' 'scripts/**/*.md'
 MDCHECKFILES := CODE_OF_CONDUCT.md CONTRIBUTING.md DEVELOPING.md README.md
+SPARSEZOO_TEST_MODE := "true"
 
 BUILD_ARGS :=  # set nightly to build nightly release
 TARGETS := ""  # targets for running pytests: keras,onnx,pytorch,pytorch_models,pytorch_datasets,tensorflow_v1,tensorflow_v1_models,tensorflow_v1_datasets

src/sparseml/framework/info.py

@@ -237,7 +237,7 @@ def save_framework_info(framework: Any, path: Optional[str] = None):
             "saved framework info for framework %s in file at %s", framework, path
         ),
     else:
-        print(info.json())
+        print(info.json(indent=4))
         _LOGGER.info("printed out framework info for framework %s", framework)
 
 

src/sparseml/sparsification/info.py

@@ -241,7 +241,7 @@ def save_sparsification_info(framework: Any, path: Optional[str] = None):
             "saved sparsification info for framework %s in file at %s", framework, path
         ),
     else:
-        print(info.json())
+        print(info.json(indent=4))
         _LOGGER.info("printed out sparsification info for framework %s", framework)