final comments

Author: rafaelpierrehf

makefile

@@ -5,7 +5,7 @@ check_dirs := src
 # run tests
 
 unit-test:
-	python3 -m pytest -s -v tests/unit -n 10 --log-cli-level='DEBUG'
+	RUN_SLOW=True python3 -m pytest -s -v tests/unit -n 10 --log-cli-level='ERROR'
 
 integ-test:
 	python3 -m pytest -s -v tests/integ/

src/huggingface_inference_toolkit/utils.py

@@ -212,7 +212,6 @@ def get_device():
     The get device function will return the device for the DL Framework.
     """
     gpu = _is_gpu_available()
-    logger.info(f"GPU Available: {gpu}")
 
     if gpu:
         return 0

tests/unit/test_serializer.py

@@ -9,7 +9,6 @@
     Imager
 )
 from PIL import Image
-import logging
 
 def test_json_serialization():
     t = {"res": np.array([2.0]), "text": "I like you.", "float": 1.2}