Skip to content

Commit 3a2c10e

Browse files
committed
fix for pylint
1 parent ea6e177 commit 3a2c10e

File tree

1 file changed

+10
-9
lines changed

1 file changed

+10
-9
lines changed

src/sagemaker/feature_store/feature_group.py

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -276,12 +276,12 @@ def _run_multi_process(self, data_frame: DataFrame, wait=True, timeout=None):
276276
for i in range(self.max_processes):
277277
start_index = min(i * batch_size, data_frame.shape[0])
278278
end_index = min(i * batch_size + batch_size, data_frame.shape[0])
279-
args += [\
280-
(self.max_workers,\
281-
self.feature_group_name, \
282-
self.sagemaker_fs_runtime_client_config, \
283-
data_frame[start_index:end_index], \
284-
start_index, \
279+
args += [(
280+
self.max_workers,
281+
self.feature_group_name,
282+
self.sagemaker_fs_runtime_client_config,
283+
data_frame[start_index:end_index],
284+
start_index,
285285
timeout)]
286286

287287
def init_worker():
@@ -297,9 +297,10 @@ def init_worker():
297297
if wait:
298298
self.wait(timeout=timeout)
299299

300-
def _run_multi_threaded(max_workers: int, feature_group_name: str, \
301-
sagemaker_fs_runtime_client_config: Config, data_frame: DataFrame, row_offset=0, timeout=None) \
302-
-> List[int]:
300+
def _run_multi_threaded(
301+
max_workers: int, feature_group_name: str,
302+
sagemaker_fs_runtime_client_config: Config,
303+
data_frame: DataFrame, row_offset=0, timeout=None) -> List[int]:
303304
"""Start the ingestion process.
304305
305306
Args:

0 commit comments

Comments
 (0)