Update black formatting

Author: qubvel

.pre-commit-config.yaml

@@ -1,6 +1,6 @@
 repos:
   - repo: https://github.com/psf/black
-    rev: 21.12b0
+    rev: 22.3.0
     hooks:
       - id: black
         args: [ --config=pyproject.toml ]

README.md

@@ -429,7 +429,7 @@ $ pip install git+https://github.com/qubvel/segmentation_models.pytorch
 
 ##### Install linting and formatting pre-commit hooks
 ```bash
-pip install pre-commit black flake8
+pip install pre-commit black==22.3.0 flake8==4.0.1
 pre-commit install
 ```
 

segmentation_models_pytorch/encoders/_base.py

@@ -21,7 +21,7 @@ def out_channels(self):
 
     @property
     def output_stride(self):
-        return min(self._output_stride, 2 ** self._depth)
+        return min(self._output_stride, 2**self._depth)
 
     def set_in_channels(self, in_channels, pretrained=True):
         """Change first convolution channels"""

segmentation_models_pytorch/encoders/timm_universal.py

@@ -39,4 +39,4 @@ def out_channels(self):
 
     @property
     def output_stride(self):
-        return min(self._output_stride, 2 ** self._depth)
+        return min(self._output_stride, 2**self._depth)

segmentation_models_pytorch/metrics/functional.py

@@ -300,8 +300,8 @@ def _compute_metric(
 
 
 def _fbeta_score(tp, fp, fn, tn, beta=1):
-    beta_tp = (1 + beta ** 2) * tp
-    beta_fn = (beta ** 2) * fn
+    beta_tp = (1 + beta**2) * tp
+    beta_fn = (beta**2) * fn
     score = beta_tp / (beta_tp + beta_fn + fp)
     return score
 

segmentation_models_pytorch/utils/functional.py

@@ -58,7 +58,7 @@ def f_score(pr, gt, beta=1, eps=1e-7, threshold=None, ignore_channels=None):
     fp = torch.sum(pr) - tp
     fn = torch.sum(gt) - tp
 
-    score = ((1 + beta ** 2) * tp + eps) / ((1 + beta ** 2) * tp + beta ** 2 * fn + fp + eps)
+    score = ((1 + beta**2) * tp + eps) / ((1 + beta**2) * tp + beta**2 * fn + fp + eps)
 
     return score
 

setup.py

@@ -35,7 +35,16 @@
     REQUIRED = []
 
 # What packages are optional?
-EXTRAS = {"test": ["pytest", "mock", "pre-commit", "flake8==4.0.1", "flake8-docstrings==1.6.0"]}
+EXTRAS = {
+    "test": [
+        "pytest",
+        "mock",
+        "pre-commit",
+        "black==22.3.0",
+        "flake8==4.0.1",
+        "flake8-docstrings==1.6.0",
+    ],
+}
 
 # Import the README and use it as the long-description.
 # Note: this will only work if 'README.md' is present in your MANIFEST.in file!