linter

Author: kboroszko

tensorflow_io/core/BUILD

@@ -180,10 +180,10 @@ cc_library(
     srcs = [
         "kernels/bigtable/bigtable_dataset_kernel.cc",
         "kernels/bigtable/bigtable_resource_kernel.h",
-        "kernels/bigtable/bigtable_row_range.h",
-        "kernels/bigtable/bigtable_row_set.h",
         "kernels/bigtable/bigtable_row_range.cc",
+        "kernels/bigtable/bigtable_row_range.h",
         "kernels/bigtable/bigtable_row_set.cc",
+        "kernels/bigtable/bigtable_row_set.h",
         "kernels/bigtable/bigtable_version_filters.cc",
         "kernels/bigtable/bigtable_version_filters.h",
         "ops/bigtable_ops.cc",

tensorflow_io/python/ops/bigtable/bigtable_dataset_ops.py

@@ -16,9 +16,7 @@ class BigtableClient:
 
     def __init__(self, project_id: str, instance_id: str):
         """Creates a BigtableClient to start Bigtable read sessions."""
-        self._client_resource = core_ops.bigtable_client(
-            project_id, instance_id
-        )
+        self._client_resource = core_ops.bigtable_client(project_id, instance_id)
 
     def get_table(self, table_id):
         return BigtableTable(self._client_resource, table_id)
@@ -29,7 +27,9 @@ def __init__(self, client_resource, table_id: str):
         self._table_id = table_id
         self._client_resource = client_resource
 
-    def read_rows(self, columns: List[str], filter: filters.BigtableFilter = filters.latest()):
+    def read_rows(
+        self, columns: List[str], filter: filters.BigtableFilter = filters.latest()
+    ):
         return _BigtableDataset(self._client_resource, self._table_id, columns, filter)
 
 
@@ -40,9 +40,7 @@ def __init__(self, client_resource, table_id: str, columns: List[str], filter):
         self._table_id = table_id
         self._columns = columns
         self._filter = filter
-        self._element_spec = tf.TensorSpec(
-            shape=[len(columns)], dtype=dtypes.string
-        )
+        self._element_spec = tf.TensorSpec(shape=[len(columns)], dtype=dtypes.string)
 
         variant_tensor = core_ops.bigtable_dataset(
             client_resource, filter._impl, table_id, columns

tensorflow_io/python/ops/bigtable/bigtable_row_set.py

@@ -31,9 +31,7 @@ def append(self, row_or_range):
         if isinstance(row_or_range, str):
             core_ops.bigtable_row_set_append_row(self._impl, row_or_range)
         else:
-            core_ops.bigtable_row_set_append_row_range(
-                self._impl, row_or_range._impl
-            )
+            core_ops.bigtable_row_set_append_row_range(self._impl, row_or_range._impl)
 
 
 def empty():
@@ -72,6 +70,4 @@ def intersect(row_set: RowSet, row_range: bigtable_row_range.RowRange):
     Returns:
       RowSet: an intersection of the given row set and row range.
     """
-    return RowSet(
-        core_ops.bigtable_row_set_intersect(row_set._impl, row_range._impl)
-    )
+    return RowSet(core_ops.bigtable_row_set_intersect(row_set._impl, row_range._impl))

tests/test_bigtable/bigtable_emulator.py

@@ -67,9 +67,7 @@ def _get_cbt_emulator_path():
 
 
 def _get_cbt_cli_path():
-    return _get_cbt_binary_path(
-        CBT_CLI_PATH_ENV_VAR, CBT_CLI_SEARCH_PATHS, "cbt cli"
-    )
+    return _get_cbt_binary_path(CBT_CLI_PATH_ENV_VAR, CBT_CLI_SEARCH_PATHS, "cbt cli")
 
 
 def _extract_emulator_addr_from_output(emulator_output):
@@ -82,9 +80,7 @@ def _extract_emulator_addr_from_output(emulator_output):
             for word in words:
                 if re.fullmatch("[a-z.0-9]+:[0-9]+", word):
                     return word
-            raise RuntimeError(
-                f"Failed to find CBT emulator in the line {line}"
-            )
+            raise RuntimeError(f"Failed to find CBT emulator in the line {line}")
 
 
 class BigtableEmulator:

tests/test_bigtable/test_read_rows.py

@@ -19,9 +19,7 @@
 
 import os
 from .bigtable_emulator import BigtableEmulator
-from tensorflow_io.python.ops.bigtable.bigtable_dataset_ops import (
-    BigtableClient,
-)
+from tensorflow_io.python.ops.bigtable.bigtable_dataset_ops import BigtableClient
 import tensorflow as tf
 from tensorflow import test
 

tests/test_bigtable/test_row_set.py

@@ -104,14 +104,10 @@ def test_from_rows_or_ranges(self):
         self.assertEqual(expected, repr(r_set))
 
     def test_intersect(self):
-        r_set = row_set.from_rows_or_ranges(
-            row_range.open_range("row1", "row5")
-        )
+        r_set = row_set.from_rows_or_ranges(row_range.open_range("row1", "row5"))
         r_set = row_set.intersect(r_set, row_range.closed_range("row3", "row7"))
         expected = (
-            "row_ranges {\n"
-            + '  start_key_closed: "row3"\n'
-            + "  end_key_open: "
+            "row_ranges {\n" + '  start_key_closed: "row3"\n' + "  end_key_open: "
             '"row5"\n' + "}\n"
         )
         self.assertEqual(expected, repr(r_set))