Enable info logging on test-cpu

autonomi-ai · Jan 24, 2024 · 728abba · 728abba
1 parent 018b743
commit 728abba
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/Makefile b/Makefile
@@ -92,7 +92,7 @@ test: ## Basic CPU testing with mkdocs-build
 
 test-cpu: ## Basic CPU testing
 	CUDA_VISIBLE_DEVICES="" \
-	pytest -sv tests
+	pytest --log-cli-level=INFO -sv tests
 
 test-gpu: ## Basic GPU testing with single GPU
 	CUDA_VISIBLE_DEVICES="0" \

diff --git a/nos/common/spec.py b/nos/common/spec.py
@@ -423,7 +423,11 @@ def load_profile_catalog(self) -> "ModelSpecMetadataCatalog":
         if not NOS_PROFILE_CATALOG_PATH.exists():
             raise FileNotFoundError(f"Model metadata catalog not found, path={NOS_PROFILE_CATALOG_PATH}.")
 
-        print("Loading profiling catalog from ", NOS_PROFILE_CATALOG_PATH, "...")
+        import logging
+
+        logger = logging.getLogger(__name__)
+        debug_str = "Loading profiling catalog from " + str(NOS_PROFILE_CATALOG_PATH)
+        logger.info(debug_str)
 
         # Read the catalog
         df = pd.read_json(str(NOS_PROFILE_CATALOG_PATH), orient="records")