black

.pre-commit-config.yaml

@@ -1,6 +1,6 @@
 repos:
   - repo: https://github.com/ambv/black
-    rev: 23.1.0
+    rev: 24.1.0
     hooks:
       - id: black
         #language_version: python3.6 

ktrain/imports.py

@@ -10,9 +10,9 @@
 
 from packaging import version
 
-os.environ[
-    "NUMEXPR_MAX_THREADS"
-] = "8"  # suppress warning from NumExpr on machines with many CPUs
+os.environ["NUMEXPR_MAX_THREADS"] = (
+    "8"  # suppress warning from NumExpr on machines with many CPUs
+)
 
 # TensorFlow
 SUPPRESS_DEP_WARNINGS = strtobool(os.environ.get("SUPPRESS_DEP_WARNINGS", "1"))

ktrain/lroptimize/optimization.py

@@ -346,14 +346,16 @@ def __call__(self, gradients):
             _ = self.step  # Create the step variable.
             self._gradients.extend(
                 [
-                    tf.Variable(
-                        tf.zeros_like(gradient),
-                        trainable=False,
-                        synchronization=tf.VariableSynchronization.ON_READ,
-                        aggregation=tf.VariableAggregation.ONLY_FIRST_REPLICA,
+                    (
+                        tf.Variable(
+                            tf.zeros_like(gradient),
+                            trainable=False,
+                            synchronization=tf.VariableSynchronization.ON_READ,
+                            aggregation=tf.VariableAggregation.ONLY_FIRST_REPLICA,
+                        )
+                        if gradient is not None
+                        else gradient
                     )
-                    if gradient is not None
-                    else gradient
                     for gradient in gradients
                 ]
             )

ktrain/text/ner/anago/callbacks.py

@@ -1,6 +1,7 @@
 """
 Custom callbacks.
 """
+
 from ....imports import *
 from .. import metrics
 

ktrain/text/ner/anago/models.py

@@ -1,6 +1,7 @@
 """
 Model definition.
 """
+
 from .... import utils as U
 from ....imports import *
 

ktrain/text/ner/anago/tagger.py

@@ -1,6 +1,7 @@
 """
 Model API.
 """
+
 from ....imports import *
 from .. import metrics
 

ktrain/text/ner/anago/utils.py

@@ -2,7 +2,6 @@
 Utility functions.
 """
 
-
 from ....imports import *
 
 

ktrain/text/qa/qa_finetuner.py

@@ -69,9 +69,11 @@ def convert_dataset_for_tensorflow(
 
     def densify_ragged_batch(features, label=None):
         features = {
-            feature: ragged_tensor.to_tensor(shape=batch_shape[feature])
-            if feature in tensor_keys
-            else ragged_tensor
+            feature: (
+                ragged_tensor.to_tensor(shape=batch_shape[feature])
+                if feature in tensor_keys
+                else ragged_tensor
+            )
             for feature, ragged_tensor in features.items()
         }
         if label is None: