make style

optimum/graphcore/custom_ops/sdk_version_hash/sdk_version_hash.py

@@ -12,20 +12,15 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
-import cppimport.import_hook
-import warnings
-import logging
+
+import cppimport.import_hook  # noqa: F401
+
 
 __all__ = ["sdk_version_hash"]
 
 
 def sdk_version_hash() -> str:
     """Graphcore SDK version hash (sanitised output from C++ function `poplar::packageHash`)"""
-    try:
-        from . import sdk_version_hash_lib
-        return sdk_version_hash_lib.sdk_version_hash()
-    except SystemExit as error:
-        warn_str = f"Failed to collect Poplar SDK package hash. Failed with error: {error}"
-        logging.warning(warn_str)
-        warnings.warn(warn_str)
-        return "unknown"
+    from . import sdk_version_hash_lib
+
+    return sdk_version_hash_lib.sdk_version_hash()

optimum/graphcore/models/whisper/modeling_whisper.py

@@ -492,6 +492,7 @@ def change_lm_head(self, restore: bool, use_cache: bool = None):
     def quantize_linear_layers(self, restore: bool, num_groups: int = 16):
         if not restore:
             from ...quantization.group_quantize import GroupQuantLinear
+
             logger.info("Group quantizing linear layers")
             for module in self.model.encoder.layers:
                 module.self_attn.q_proj = GroupQuantLinear.from_model(module.self_attn.q_proj, num_groups)

optimum/graphcore/quantization/__init__.py

@@ -12,8 +12,10 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 from pathlib import Path
+
 from ..custom_ops.utils import load_lib
 
+
 load_lib(Path(__file__).parent.parent / "custom_ops/group_quantize_decompress/group_quantize_decompress.cpp")
 
-from .group_quantize import group_quantize_compress, group_quantize_decompress, GroupQuantLinear
+from .group_quantize import GroupQuantLinear, group_quantize_compress, group_quantize_decompress

tests/test_custom_ops.py

@@ -17,12 +17,12 @@
 
 
 class TestCustomOps(unittest.TestCase):
-
     def test_group_quantize_compiles(self):
         # Importing quantize should compile
-        from optimum.graphcore.quantization import group_quantize
+        from optimum.graphcore.quantization import group_quantize  # noqa: F401
 
     def test_sdk_version_hash(self):
-        from optimum.graphcore.custom_ops.sdk_version_hash import sdk_version_hash 
+        from optimum.graphcore.custom_ops.sdk_version_hash import sdk_version_hash
+
         hash = sdk_version_hash()
         self.assertTrue(type(hash) is str)