fmt

Signed-off-by: Thara Palanivel <[email protected]>

Author: tharapalanivel

fms_mo/utils/dq_utils.py

@@ -38,7 +38,9 @@ def config_quantize_smooth_layers(qcfg):
         "granite-20b-code",
         "granite-20b-code",
     ]
-    if any(model in qcfg["model"] for model in llama_architecture) or any(model in qcfg["model_type"] for model in llama_architecture):
+    if any(model in qcfg["model"] for model in llama_architecture) or any(
+        model in qcfg["model_type"] for model in llama_architecture
+    ):
         qcfg["qlayer_name_pattern"] = ["model.layers."]
         qcfg["scale_layers"] = ["k_proj", "v_proj", "gate_proj", "up_proj"]
         qcfg["qskip_layer_name"] = []

fms_mo/utils/logging_utils.py

@@ -40,7 +40,7 @@ def set_log_level(log_level=None, logger_name=None):
     # Configure Python native logger
     # If CLI arg is passed, assign same log level to python native logger
     log_level = log_level or os.environ.get("LOG_LEVEL", "WARNING")
-    
+
     logging.basicConfig(
         format="%(levelname)s:%(filename)s:%(message)s", level=log_level.upper()
     )

tests/artifacts/testdata/__init__.py

@@ -12,8 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""Helpful datasets for configuring individual unit tests.
-"""
+"""Helpful datasets for configuring individual unit tests."""
+
 # Standard
 import os
 

tests/test_run_quant.py

@@ -34,9 +34,7 @@
 )
 from tests.artifacts.testdata import MODEL_NAME, WIKITEXT_TOKENIZED_DATA_JSON
 
-MODEL_ARGS = ModelArguments(
-    model_name_or_path=MODEL_NAME, torch_dtype="float16"
-)
+MODEL_ARGS = ModelArguments(model_name_or_path=MODEL_NAME, torch_dtype="float16")
 DATA_ARGS = DataArguments(
     training_data_path=WIKITEXT_TOKENIZED_DATA_JSON,
 )
@@ -73,7 +71,7 @@ def test_run_train_fails_training_data_path_not_exist():
     """Check fails when data path not found."""
     updated_data_path_args = copy.deepcopy(DATA_ARGS)
     updated_data_path_args.training_data_path = "fake/path"
-    with pytest.raises(FileNotFoundError):  # TPP Should this be FileNotFoundError or ValueError?
+    with pytest.raises(FileNotFoundError):
         quantize(
             model_args=MODEL_ARGS,
             data_args=updated_data_path_args,