try to make codestyle checker happy

apache · Jul 18, 2020 · 42d5091 · 42d5091
1 parent 9ff2f8d
commit 42d5091
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/submarine-sdk/pysubmarine/submarine/ml/pytorch/input/libsvm_dataset.py b/submarine-sdk/pysubmarine/submarine/ml/pytorch/input/libsvm_dataset.py
@@ -85,7 +85,7 @@ def _locate_sample_offsets(cls, data_uri: str, n_jobs: int) -> np.ndarray:
     @classmethod
     def _locate_sample_offsets_job(
             cls, data_uri: str, task: Tuple[int, Tuple[int, int]]) -> List[int]:
-        job_id, (start, end) = task
+        _, (start, end) = task
         offsets = [start]
         with open_buffered_file_reader(data_uri) as infile:
             infile.seek(start, os.SEEK_SET)

diff --git a/submarine-sdk/pysubmarine/submarine/ml/pytorch/layers/core.py b/submarine-sdk/pysubmarine/submarine/ml/pytorch/layers/core.py
@@ -13,8 +13,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from itertools import accumulate
-
 import torch
 from torch import nn
 

diff --git a/submarine-sdk/pysubmarine/submarine/utils/fileio.py b/submarine-sdk/pysubmarine/submarine/utils/fileio.py
@@ -14,7 +14,6 @@
 # limitations under the License.
 
 from pyarrow import fs
-from pyarrow.lib import NativeFile
 
 import io
 from urllib.parse import urlparse
@@ -52,12 +51,12 @@ def write_file(buffer: io.BytesIO,
         output_stream.write(buffer.getbuffer())
 
 
-def open_input_file(uri: str) -> NativeFile:
+def open_input_file(uri: str):
     filesystem, path = _parse_uri(uri)
     return filesystem.open_input_file(path)
 
 
-def open_output_stream(uri: str) -> NativeFile:
+def open_output_stream(uri: str):
     filesystem, path = _parse_uri(uri)
     return filesystem.open_output_stream(path)