Lint

Author: polyaxon-ci

traceml/tests/test_events_processing/test_df_processor.py

@@ -51,11 +51,7 @@ def setUp(self):
                 dbool2=np.random.choice(["a", "b"], size=self.size),
                 duniques=["x{}".format(i) for i in range(self.size)],
                 dcategoricals=[
-                    "a".format(i)
-                    if i % 2 == 0
-                    else "b".format(i)
-                    if i % 3 == 0
-                    else "c".format(i)
+                    "a" if i % 2 == 0 else "b" if i % 3 == 0 else "c"
                     for i in range(self.size)
                 ],
                 dnumerics1=range(self.size),

traceml/tests/test_summary/test_dfsummary.py

@@ -50,11 +50,7 @@ def setUp(self):
                 dbool2=np.random.choice(["a", "b"], size=self.size),
                 duniques=["x{}".format(i) for i in range(self.size)],
                 dcategoricals=[
-                    "a".format(i)
-                    if i % 2 == 0
-                    else "b".format(i)
-                    if i % 3 == 0
-                    else "c".format(i)
+                    "a" if i % 2 == 0 else "b" if i % 3 == 0 else "c"
                     for i in range(self.size)
                 ],
                 dnumerics1=range(self.size),

traceml/traceml/processors/events_processors/events_image_processors.py

@@ -311,15 +311,11 @@ def convert_to_HWC(tensor, input_format):  # noqa
     assert len(set(input_format)) == len(
         input_format
     ), "You can not use the same dimension shorthand twice. \
-        input_format: {}".format(
-        input_format
-    )
+        input_format: {}".format(input_format)
     assert len(tensor.shape) == len(
         input_format
     ), "size of input tensor and input format are different. \
-        tensor shape: {}, input_format: {}".format(
-        tensor.shape, input_format
-    )
+        tensor shape: {}, input_format: {}".format(tensor.shape, input_format)
     input_format = input_format.upper()
 
     if len(input_format) == 4:

traceml/traceml/serialization/base.py

@@ -1,6 +1,6 @@
 import os
 
-from typing import Dict, List
+from typing import List
 
 from clipped.utils.enums import get_enum_value
 from clipped.utils.paths import check_or_create_path
@@ -21,7 +21,7 @@ class EventWriter:
     def __init__(self, run_path: str, backend: str):
         self._events_backend = backend
         self._run_path = run_path
-        self._files = {}  # type: Dict[str, LoggedEventListSpec]
+        self._files = {}  # type: dict[str, LoggedEventListSpec]
         self._closed = False
 
     def _get_event_path(self, kind: str, name: str) -> str: