add pylint ingore statements

epistoteles · Jun 1, 2024 · bda737b · bda737b
1 parent 27d551c
commit bda737b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tensorhue/__init__.py b/tensorhue/__init__.py
@@ -14,7 +14,9 @@ def viz(self) -> None:
     Prints the tensor using a colored Unicode art representation.
     This method checks the type of the tensor and calls the `_viz_tensor` function with the appropriate colors.
     """
-    if isinstance(self, torch.FloatTensor):
+    if isinstance(
+        self, torch.FloatTensor
+    ):  # pylint: disable=possibly-used-before-assignment
         _viz_tensor(self)
     elif isinstance(self, torch.BoolTensor):
         _viz_tensor(self, (COLORS["false"], COLORS["true"]))

diff --git a/tensorhue/colors.py b/tensorhue/colors.py
@@ -12,7 +12,7 @@
     "white": ColorTriplet(255, 255, 255),  # white
 }
 
-COLOR_MAPPING = {
+COLOR_MAPPING = {  # pylint: disable=consider-using-namedtuple-or-dataclass
     "default": ("default_dark", "default_bright"),
     "bool": ("false", "true"),
 }